用 Python 编写的程序，用于查找已排序数组中的绝对差值之和

pythonserver side programmingprogramming更新于 2025/12/2 3:56:17

假设我们有一个数组 nums，并且按非递减顺序排序。我们必须创建一个名为 result 的数组，其长度与 nums 相同，使得 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

为了解决这个问题，我们将遵循以下步骤 −

res := 一个新列表
s:= 0
n := nums 的大小
对于范围从 1 到 n - 1 的 i，执行
- s := s + nums[i] - nums[0]
在 res 的末尾插入 s
对于范围从 1 到 n - 1 的 i，执行
- diff := nums[i] - nums[i-1]
- s := s + diff*i
- s := s - diff *(n-i)
在 res 末尾插入 s
返回 res

示例

让我们看看下面的实现以便更好地理解 −


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))