在Python中找到给定数组中的所有良好索引
假设我们有一个数字数组A，我们必须找到该数组的所有索引，以便从数组中删除第ith个元素后，该数组将成为一个好的数组。我们必须记住-
好的数组是一个元素等于所有其他元素之和的数组。
基于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：=k/2
如果k在my_map中，则
显示我+1
如果（A[i]与k相同，且my_map[k]>1]或（A[i]与k不相同），则
k：=加-A[i]
如果kmod2与0相同，则
示例
让我们看下面的实现以更好地理解-
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