给定数组中其索引的 K 次方的元素计数

在给定一个整数数组和一个整数K的情况下，求出该数组中元素的索引的 K 次方之和。

例如，给定数组[1,2,3,4]和K = 2，我们希望计算1 ^ 2 + 2 ^ 2 + 3 ^ 2 + 4 ^ 2的和，结果是30。

算法思路

算法的实现非常直观。 我们只需要对数组中的每个元素进行迭代，并将对应的索引的 k 次方添加到总和中即可。

以下是算法的伪代码：

sum = 0
for i in range(n):
   sum += arr[i] ^ k

这个算法的时间复杂度为 O(N)，其中N是数组中的元素数量。

实现代码

下面是用Python实现的代码片段：

def kPowerSum(arr, k):
   """
   给定数组中其索引的 K 次方的元素计数
   """
   n = len(arr)
   sum = 0
   for i in range(n):
       sum += arr[i] ** k
   return sum

示例

现在，我们来使用一个示例来说明如何使用这个函数：

arr = [1, 2, 3, 4]
k = 2
print(kPowerSum(arr, k))

这将输出30，说明1 ^ 2 + 2 ^ 2 + 3 ^ 2 + 4 ^ 2的和为30。

至此，我们已经学会了如何计算给定数组中其索引的 K 次方的元素计数。