python快速寻找满足条件的两个数的算法示例

前端之家收集整理的这篇文章主要介绍了python快速寻找满足条件的两个数的算法示例前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
对python这个高级语言感兴趣的小伙伴,下面一起跟随编程之家 jb51.cc的小编两巴掌来看看吧!

题目前提是一定存在这样两个数

解法一就不写了...一般想不到吧

一开始想到的是解法二最后的用hash表

(其实是想到创建一个跟target一样大的数组啦..存在就写入index,但是要全部找出,那得二维数组,但是后面想到target要是很大的话,是不是浪费空间了...所以改成Dict)

后面发现题目只要求给出两个数就好了啊- -

扩展问题比较有意思

找三个应该不难,其它还不清楚,有想再补充...

1.二维数组


# @param python算法 - 快速寻找满足条件的两个数
# @author 编程之家 jb51.cc|512Pic.com 

def find_pair(A,target):
    B = [[] for i in range(target + 1)]
    for i in range(0,len(A)):
        if A[i] <= target:
            B[A[i]].append(i)
    for i in range(0,target / 2 + 1):
        if len(B[i]) != 0 and len(B[target - i]) != 0:
            print(i,B[i],target-i,B[target-i])
 
if __name__ == "__main__":
    A = [0,1,2,11,8,3,4,5,6,7,9,10]
    find_pair(A,9)

# End www.jb51.cc

2.字典


# @param python算法 - 快速寻找满足条件的两个数
# @author 编程之家 jb51.cc|512Pic.com 

def find_pair(A,target):
    B = {}
    for i in range(0,len(A)):
        if A[i] <= target:
            if not B.has_key(A[i]):
                B[A[i]] = [i]
            else:
                B[A[i]].append(i)
    for i in range(0,target / 2 + 1):
        if B.has_key(i) and B.has_key(target-i):
            print(i,9)

# End www.jb51.cc

3.这种方法都已经重新排序了,不知道书上还返回索引有什么意义...排序偷懒直接用内置的啦...


# @param python算法 - 快速寻找满足条件的两个数
# @author 编程之家 jb51.cc|512Pic.com 

def find_pair(A,target):
    A.sort()
    i,j = 0,len(A) - 1
    while i < j:
        s = A[i] + A[j]
        if s == target:
            print(i,A[i],j,A[j])
            i += 1
            j -= 1
        elif s < target:
            i += 1
        else:
            j -= 1
 
if __name__ == "__main__":
    A = [0,9)

# End www.jb51.cc

猜你在找的Python相关文章