介绍

在程序开发过程中，经常需要对数组中的元素进行两两组合并计算，例如求出数组中所有元素的差之和、积之和等。为了方便，可以使用一个函数f(a[i], a[j])来表示对数组中元素的操作。那么，如何求出在n个整数数组中所有对的总和呢？

实现

先定义一个数组 a，数组长度为 n，然后使用两个嵌套循环枚举数组中所有元素的组合，计算它们的和或其他操作，并将结果累加到一个变量 res 中，最后返回 res 即可。

以下是示例伪代码：

def calculate_sum(a, n, func):
    res = 0
    for i in range(n):
        for j in range(i + 1, n):
            res += func(a[i], a[j])
    return res

其中，func 为传入的函数参数，用于对数组元素的操作。

示例

例如，我们需要计算一个数组中所有元素的差值之和，则可以定义一个函数 diff(a, b)：

def diff(a, b):
    return abs(a - b)

然后调用 calculate_sum 函数：

a = [1, 3, 5, 7, 9]
n = len(a)
res = calculate_sum(a, n, diff)
print(res)

输出结果为 20，即所有元素间的差值之和为 |1-3| + |1-5| + |1-7| + |1-9| + |3-5| + |3-7| + |3-9| + |5-7| + |5-9| + |7-9| = 20。

小结

使用函数统一表示数组元素的操作，可以使代码更加简洁、清晰；同时在需要统计数组元素两两组合的场景下，可以使用嵌套循环来实现。