在 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]

以上是 在 Python 中查找排序数组中的绝对差之和的程序 的全部内容, 来源链接: utcz.com/z/347505.html

回到顶部