Pythonで指定された配列内のすべての適切なインデックスを検索します

数値の配列Aがあるとすると、この配列のすべてのインデックスを見つけて、配列からi番目の要素を削除した後、その配列が適切な配列になるようにする必要があります。そのことを覚えておく必要があります-

• 適切な配列とは、他のすべての要素の合計に等しい要素を持つ配列です。
• 1ベースのインデックスがここで使用されます。

したがって、入力が[10、4、6、2]の場合、出力は[1,4]になり、A [1]を削除すると、配列は[4、6、2]のようになります。 6 =4 + 2なので、良いです。 A [4]を削除すると、配列は[10、4、6]のようになり、10 =4+6のようになります。

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

• n：=Aのサイズ
• 追加：=0
• my_map：=新しい地図
• 0からnの範囲のiについては、
  • my_map [A​​ [i]]：=my_map [A​​ [i]] + 1
  • 追加：=追加+ A [i]
• 0からnの範囲のiについては、
  • k：=追加-A [i]
  • k mod 2が0と同じ場合、
    • k：=k / 2
    • my_mapでkの場合、
      • （A[i]がkと同じでmy_map[k]> 1）または（A [i]がkと同じでない）の場合、
        • i+1を表示

例

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

from collections import defaultdict
def find_indices(A):
   n = len(A)
   add = 0
   my_map = defaultdict(lambda:0)
   for i in range(n):
      my_map[A[i]] += 1
      add += A[i]
   for i in range(n):
      k = add - A[i]
      if k % 2 == 0:
         k = k >> 1
         if k in my_map:
            if ((A[i] == k and my_map[k] > 1) or (A[i] != k)):
               print((i + 1))
A = [10, 4, 6, 2]
find_indices(A)

入力

[10, 4, 6, 2]

出力

1
4