在 Python 中查找排序数组中的绝对差之和的程序
假设我们有一个数组nums并且按非递减顺序排序。我们必须创建一个与nums长度相同的名为result的数组，使得result[i]是nums[i]与数组中所有其他元素之间绝对差的总和。
所以，如果输入像nums=[5,7,12]，那么输出将是[9,7,12]因为
|5-5|+|5-7|+|5-12|=0+2+7=9
|7-5|+|7-7|+|7-12|=2+0+5=7
|5-12|+|7-12|+|12-12|=7+5+0=12
示例
让我们看看以下实现以获得更好的理解-
def solve(nums):
res = []
s=0
n = len(nums)
for i in range(1,n):
s+=nums[i]-nums[0]
res.append(s)
for i in range(1,n):
diff = nums[i]-nums[i-1]
s += diff*i
s -= diff *(n-i)
res.append(s)
return res
nums = [5,7,12]
print(solve(nums))输入
[5,7,12]输出结果
[9, 7, 12]