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

当前位置:

首页 > Python基础教程 >

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

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

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

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

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

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

扩展问题比较有意思

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

1.二维数组

def find_pair(A, target):

    B= [[]for iin range(target+ 1)]

    for iin range(0,len(A)):

        if A[i] <= target:

            B[A[i]].append(i)

    for iin 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,1,2,11,8,3,4,5,6,7,8,9,10]

    find_pair(A,9)

2.字典

def find_pair(A, target):

    B= {}

    for iin 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 iin range(0, target/ 2 + 1):

        if B.has_key(i)and B.has_key(target-i):

            print(i, B[i], target-i, B[target-i])

if __name__== "__main__":

    A= [0,1,1,2,11,8,3,4,5,6,7,8,9,10]

    find_pair(A,9)

栏目列表