Pythonで公正な配列を作成する方法を見つけるためのプログラム

numsという配列があるとします。インデックスを1つだけ選択して、そのインデックスから要素を削除できます。 （要素のインデックスは、削除後に変更される場合があります）。奇数のインデックス値の合計が偶数のインデックス値の合計と等しい場合、配列は公平であると言えます。削除後、numsが公平になるように、選択できるインデックスの数を見つける必要があります。

したがって、入力がnums =[5,3,7,2]のような場合、出力は

• インデックス0から削除します。配列は[3,7,2]、偶数の位置の合計：3 + 2 =5、奇数の位置の合計7（不公平）

• インデックス1から削除すると、配列は[5,7,2]、偶数の位置の合計：5 + 2 =7、奇数の位置の合計7（公平）

• インデックス2から削除すると、配列は[5,3,2]、偶数の位置の合計：5 + 2 =7、奇数の位置の合計3（不公平）

• インデックス3から削除すると、配列は[5,3,7]、偶数の位置の合計：5 + 7 =12、奇数の位置の合計3（不公平）

これを解決するには、次の手順に従います-

• res：=0、sm1：=0、sm2：=0
• 範囲1からnums-1のサイズのiの場合、do
  • 私が偶数の場合、
    • sm1：=sm1 + nums [i]
  • それ以外の場合、
    • sm2：=sm2 + nums [i]
• sm1がsm2と同じ場合、
  • res：=res + 1
• 範囲1からnums-1のサイズのiの場合、do
  • iが奇数の場合、
    • sm1：=sm1-nums [i] + nums [i-1]
  • それ以外の場合、
    • sm2：=sm2-nums [i] + nums [i-1]
  • sm1がsm2と同じ場合、
    • res：=res + 1
• return res

例

理解を深めるために、次の実装を見てみましょう-

def solve(nums):
   res, sm1, sm2 = 0, 0, 0
   for i in range(1, len(nums)):
      if i%2 == 1:
         sm1 += nums[i]
      else:
         sm2 += nums[i]
   if sm1 == sm2:
      res += 1

   for i in range(1, len(nums)):
      if i % 2 == 1:
         sm1 = sm1 - nums[i] + nums[i-1]
      else:
         sm2 = sm2 - nums[i] + nums[i-1]
      if sm1 == sm2:
         res += 1

   return res

nums = [5,3,7,2]
print(solve(nums))

入力

[5,3,7,2]

出力

1