python Solution LeetCode No.1 TwoSum（两数之和）

84 阅读 0 评论 56 点赞

我是靠谱客的博主完美机器猫，这篇文章主要介绍python Solution LeetCode No.1 TwoSum（两数之和），现在分享给大家，希望可以做个参考。

题目:

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

复制代码

1
2
3
4
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

代码:

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution:
    def twoSum(self, nums, target):
        resultlist = []
        numsSet = set(nums)
        for m, n in enumerate(nums[:-1]):
            if target - n in numsSet:
                if nums.count(n) > 1 and target - n == n:
                    numsSet.remove(n)
                    j = nums.index(target - n, m+1)
                elif nums.count(n) < 2 and target - n == n:
                    continue
                else:
                    numsSet.remove(n)
                    numsSet.remove(target - n)
                    j = nums.index(target - n)
                resultlist.append(m)
                resultlist.append(j)
        return  resultlist

结果:

说明:

1.使用set比两重循环要节省很多时间

2.在列表循环中先进行了一步处理，判断给定的target值是否等于n+n，并且判断在原列表中出现的次数，如果是且 target-n 在列表中只出现了一次，则可以保证不存在相同值解，跳过。如果是偶数且 target除2 在列表中只出现了一次以上，则认为存在相同值的解，直接存放到结果中，并将该值从set中删除。举个例子如果输入数组是[3,3],target是6，则返回结果是[0,1]，如果不提前处理偶数的相同解情况，结果就会出现问题，因为set本身就进行了去重。