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

為了解決這個問題，我們將遵循以下步驟：

• res := 一個新的列表
• s:= 0
• n := nums 的大小
• 對於 i 從 1 到 n - 1 的範圍，執行：
  • s := s + nums[i] - nums[0]
• 在 res 的末尾插入 s
• 對於 i 從 1 到 n - 1 的範圍，執行：
  • 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))

輸入

[5,7,12]

輸出

[9, 7, 12]

Arnab Chakraborty

更新於: 2021年10月6日

442 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習