python 整数对查找

题目描述

请设计一个高效算法，找出数组中两数之和为指定值的所有整数对。

给定一个int数组A和数组大小n以及需查找的和sum，请返回和为sum的整数对的个数。保证数组大小小于等于3000。

测试样例：

[1,2,3,4,5],5,6

返回：2

# -*- coding:utf-8 -*-
import collections
class FindPair:
    def countPairs(self, A, n, tsum):
        res = 0
        setA = list(set(A))
        setA.sort()
        start , end = 0 , len(setA)-1
        dd = collections.defaultdict(int)
        for i in A :
            dd[i] += 1
        while start < end :
            if setA[start]+setA[end] < tsum :
                start += 1
            elif setA[start]+setA[end] > tsum :
                end -= 1
            else :
                res += dd[setA[start]]*dd[setA[end]]
                start += 1
                end -= 1
        if setA[start]*2 == tsum :
            res += dd[setA[start]]*(dd[setA[start]]-1)//2
        return res

思路：首先获取去除掉重复数字并且排好序的数据，然后用一个字典类型记录每个数出现的次数，再然后同时从首尾出发判断当前整数对的和与目标和的大小，最后再依此计算满足条件的整数对的数量，至于为什么这么加自己意会应该能懂？其中 collections.defaultdict(int)这个比较让楼主在意，这里的defaultdict(function_factory)构建的是一个类似dictionary的对象，其中keys的值，自行确定赋值，但是values的类型，是function_factory的类实例，而且具有默认值。比如default(int)则创建一个类似dictionary对象，里面任何的values都是int的实例，而且就算是一个不存在的key, d[key] 也有一个默认值，这个默认值是int()的默认值0。

关于defaultdict()可以参考下面两篇博客：

https://blog.csdn.net/yangsong95/article/details/82319675#commentBox

https://blog.csdn.net/kyi_zhu123/article/details/80203118