python 嵌套两个for循环出现问题?

题目要求:

给定 nums = [2, 7, 11, 15], target = 9

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

题目来源:leetcode上的sum求和

个人代码

def twoSum(nums, target):    l = []  # 外部栈存储
    s = 0  # 外部循环次数
    for _ in nums:        i = 0
        s = s + 1
        p = nums.pop(0)
        l.append(p)
        for v in nums:            if v == target - p:
                return [len(l) - 1, s + i]            i = i + 1

个人思路与问题

我的想法,是取出1个元素放入外部栈,并记录其值,然后在剩下的列表中遍历满足条件的元素.
s,i,为了标记满足要求的元素的序号作用.
验证过程中发现,每当我输入

print(twoSum([1,3,3,5,2],7))

调试结果就会显示None,具体就是当满足的两个数都在输入列表末尾,就会出现None.

python新手,请多指教.


跃然一笑
浏览 572回答 1
1回答

收到一只叮咚

先上结论,问题应该出在for _ in nums: 和nums.pop(0)同时使用。这会导致你每次迭代取的一个值是从一个不断缩小的列表里取的。你可以debugger或者把每次循环的_ 给print出来看,你会发现每次循环都跳了一次。顺便给出一个同样复杂度n^2但比较容易理解的solution:def twoSum(nums, target):     for i in range(len(nums)):         j = i + 1         tmp = nums[i] + nums[j]        if tmp == target:            return [i, j] print(twoSum([1,4,3,5,2],7))
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python