在python中获取给定数字列表中的一对数字的函数

Question

我正在编写一个函数，它试图返回一个整数，该整数表示给定列表中匹配的数字对的数量。下面是代码。

def testfunc(n, ar):
    count = []
    ar.sort()
    print(ar)
    if len(ar) == n:
        for i in range(len(ar)):
            for j in range(i+1, len(ar)):
                if ar[i] == ar[j]:
                    count.append(1)
                    if ar[i+1] == ar[j+1]:
                        count.append(1)
                        break
    return len(count)

当输入如下，测试用例 1 时有效，但在测试用例 2 时失败。

测试用例 1：

n = 9
ar = [10, 20, 20, 10, 10, 30, 50, 10, 20]

测试用例 2：

n = 10
ar = [1, 1, 3, 1, 2, 1, 3, 3, 3, 3]

请指教我做错了什么。

Answer 1

如果您计算列表中每个项目的出现次数，获取项目到计数的映射，则项目的匹配对数是该项目的计数除以 2 向下舍入。我们可以使用collections.Counter 来统计物品，并使用地板除法// 来获得每个物品的匹配对数。

from collections import Counter

def testfunc(arr):
    return sum(v//2 for v in Counter(arr).values())

print(testfunc([10, 20, 20, 10, 10, 30, 50, 10, 20])) # 3
print(testfunc([1, 1, 3, 1, 2, 1, 3, 3, 3, 3]))       # 4
print(testfunc([2,2,3,3,5,8,5,9,8]))                  # 4

无需传递列表的长度，因为您始终可以通过调用len(arr) 来获取列表的长度。

Answer 2

你在

中遍历数组长度

如果 arg[i+1] == arg[j+1]:

试试这个，它应该适用于这两种情况。

def testfunc(ar):
    ar.sort();
    count = i = 0
    while i + count < len(ar) - 1:
        if ar[count + i] == ar[count + i + 1]:
            count += 1
        i += 1
    return count