Pythonでソートされた配列の絶対差の合計を見つけるプログラム

配列numsがあり、それが降順ではない順序でソートされているとします。 result [i]がnums[i]と配列内の他のすべての要素との絶対差の合計になるように、numsと同じ長さのresultという配列を作成する必要があります。

したがって、入力が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の場合、do
  • s：=s + nums [i]-nums [0]
• resの最後にsを挿入
• 1からn-1の範囲のiの場合、do
  • diff：=nums [i]-nums [i-1]
  • s：=s + diff * i
  • s：=s --diff *（n-i）
• resの最後にsを挿入
• return 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]