我想找出重复元素数组中元素的索引

Question

a=[2, 1, 3, 5, 3, 2]def firstDuplicate(a):    for i in range(0,len(a)):        for j in range(i+1,len(a)):                            while a[i]==a[j]:                num=[j]                break    print(num)print(firstDuplicate(a))输出应该是4和5，但是只有4

Answer 1

您可以在O（n）时间和O（1）多余空间中找到数组中所有重复项的索引，如下所示：

def get_duplicate_indices(arr):

    inds = []

    for i, val in enumerate(arr):

        val = abs(val)

        if arr[val] >= 0:

            arr[val] = -arr[val]

        else:

            inds.append(i)

    return inds

get_duplicate_indices(a)

[4, 5]

注意，这将修改数组！如果要保持输入数组不变，则将上面的前几行替换为：

def get_duplicate_indices(a):

    arr = a.copy()  # so we don't modify in place. Drawback is it's not O(n) extra space

    inds = []

    for i, val in enumerate(a):

        # ...

从本质上讲，这使用数组中每个元素的符号作为以前是否看到过数字的指示器。如果我们遇到一个负值，则意味着我们已经看过达到的数字，因此我们将数字的索引附加到我们已经看到的索引列表中。

请注意，如果数组中的值大于数组的长度，则可能会遇到麻烦，但是在这种情况下，我们只是将工作数组扩展为与输入中的最大值相同的长度。十分简单。

Answer 2

您的代码有些错误。以下将收集每个第一个重复项的索引：

def firstDuplicate(a):

    num = []  # list to collect indexes of first dupes

    for i in range(len(a)-1):  # second to last is enough here

        for j in range(i+1, len(a)):                

            if a[i]==a[j]:  # while-loop made little sense

                num.append(j)  # grow list and do not override it

                break  # stop inner loop after first duplicate

    print(num)

当然，还有更多性能算法可以实现这一点，而不是二次方的。