如何从数组中找到满足a + b = c + d的两个数对

class pair:
    def __init__(self, first, second):
        self.first = first
        self.second = second


def findPairs(arr):
    # 键为数对的和, 值为对
    sumPair = dict()
    n = len(arr)
    # 遍历数组中所有可能的数对
    i = 0
    while i < n:
        j = i + 1
        while j < n:
            # 如果这个数对的和在map中没有,则放入map中
            sums = arr[i] + arr[j]
            if sums not in sumPair:
                sumPair[sums] = pair(i, j)
            else:
                # 找出已经遍历过的并存储在map中和为sum的数对
                p = sumPair[sums]
                print('(' + str(arr[p.first]) + ',' + str(arr[p.second]) + ')', '(' + str(arr[i]) + ',' + str(arr[j]) + ')')
                return True

            j += 1
        i += 1
    return False


if __name__ == '__main__':
    arr = [3, 4, 7, 10, 20, 9, 8]
    findPairs(arr)

运行结果如下:
(3,8) (4,7)

你可能感兴趣的:(python数据结构和算法)