LeetCode 两数之和

题目 1： 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例：

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

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

暴力解法

暴力法很简单，遍历每个元素 $x$，并查找是否存在一个值与 $target-x$ 相等的目标元素。这种方法的时间复杂度为 $O(n^2)$，空间复杂度为 $O(1)$，C++ 的代码如下：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        for(int i = 0; i < nums.size(); i++)
        {
            for(int j = i + 1; j < nums.size(); j++)
            {
                if(nums[i] + nums[j] == target)
                {
                    return {i,j};
                }
            }
        }
        return {};
    }
};

将上述代码写成 python 语言如下：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            for j in range(i + 1, len(nums)):
                if nums[i] + nums[j] == target:
                    return [i, j]
        return []

上述两个程序运行情况如下：

使用 python 语言所需要的时间明显比 C++ 所需要的时间要多很多倍，这可能是语言本身的可能。可以做的改变是将第二个循环使用 list 之间的运算和关键词 in 来检查是否满足要求。具体代码如下：

class Solution: 
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        lens = len(nums)
        j = -1
        for i in range(1, lens):
            temp = nums[:i]
            if (target - nums[i]) in temp:
                j = temp.index(target - nums[i])
                break
        if j >= 0:
            return [j, i]

运行情况如下：

明显比原来的 python 运行的时间缩短了很多。