python列表编程题

题目要求

自定义函数twonums_sum(n, lst)，在列表lst中查找是否有两数之和等于n,若有则返回两数的下标，否则返回-1。对于一个不包含重复数字的有序列表[1, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 15, 18, 19, 20, 21, 29, 34, 54, 65],从键盘输入n,调用函数twonums_sum() 输出满足条件的两个数的下标（找到一组即可且要求其中的一个数尽量小)，若所有数均不满足条件则输出“not found”。
[输入样例]
17
[输出样例]
(1，10)

代码

def twonums_sum(n, lst):
    for k in range(len(lst)):
        flag = 0
        for j in range(len(lst)):
            if lst[k] + lst[j] == n:
                print((k, j))
                flag = 1
                break
        if flag == 1:
            break
        if k == (len(lst)-1) and flag == 0:
            print("not found")


if __name__ == '__main__':
    lst = [1, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 15, 18, 19, 20, 21, 29, 34, 54, 65]
    n = eval(input())
    twonums_sum(n, lst)

运行结果