码迷,mamicode.com
首页 > 编程语言 > 详细

求数组两个数的和的索引

时间:2016-03-23 22:13:49      阅读:193      评论:0      收藏:0      [点我收藏+]

标签:

做了一道题, 有一个有序数组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)] 正确

  

求数组两个数的和的索引

标签:

原文地址:http://www.cnblogs.com/wuweixin/p/5313076.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!