求数组两个数的和的索引

Posted 2020-06-28 Wyshon

做了一道题， 有一个有序数组a，包含n个元素[a1,a2,...,aN]，找出相加等于x的两个数的下标。

我看到这道题，第一个想法是，这两个数肯定不止一样的，不管它要求是返回一个就行还是返回所有，我都返回所有。我用python语言实现，借鉴二分查找法，复杂度nlogn，不知道有没大神有更好的解法，我只能做到这了。代码如下：

__author__ = ‘weixin‘
#-*- coding:utf-8 -*-
def findAdd(list,num):
    length = len(list)
    indexList = []
    for index,value in enumerate(list):
        start = index
        end = length
        while start<end:
            mid = int((start+end)/2)
            if list[mid]+value < num:
                start = mid+1
            else:
                end = mid
        if list[start] + value == num and (start != index):
            indexList.append((start,index))
    return indexList

#测试代码
list = [1,2,2,3,4,5,6,7,8]
print findAdd(list,7)

执行结果
[(6, 0), (5, 1), (5, 2), (4, 3)] 正确