LeetCode刷题之求两数之和

题目:
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解决思路:
1.遍历两次列表,然后判断遍历得到的数之和是否等于目标值,是:返回,否:继续遍历
2.遍历一次列表,利用减法,判断另一个数是否在列表中,是:返回,否:继续。
二者对比,第一种思路遍历两次,时间复杂度比第二种高,故采用第二种解法。

l = []
def two_sum(nums, target):
    n = len(nums)
    for i in range(n):
        a = target - nums[i]
        if a in nums[i + 1::]:
            b = nums.index(a)
            l.append(i)
            l.append(b)
            return l
        else:
            continue


nums = [2, 7, 11, 19]
ret = two_sum(nums, 9)
print(ret)

猜你喜欢

转载自blog.csdn.net/cestlavieqiang/article/details/83717469