求数组两个数的和的索引
做了一道题, 有一个有序数组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)] 正确