LeetCode刷题记录1-两数之和

题目

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

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

示例：

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

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

普通解法 时间O(n^2)，空间O(1)

这可能是大家最快能想到的方法了，直接列出来(简单粗暴双重循环)：

vector<int> twoSum(vector<int>& nums, int target) {
    if (nums.empty())
    {
        return{};
    }
    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{};
}

LeetCode运行结果如下：

Hash表的空间换时间方式 时间O(n)，空间O(n)

vector<int> twoSum(vector<int> &numbers, int target)
{
    //map to index and vector value
    //Key is the number and value is its index in the vector.
    unordered_map<int, int> hash;  

    for (int i = 0; i < numbers.size(); i++) {
        int res = target - numbers[i];

        //原理就是:
        /*
         需要求a+b = 13 => 2 + 11 = 13
         1. numbers[i] = 2 则 res = 11 此时11不存在于数组将此时的{2,0}记录
         2. numbers[i] = 7 则 res = 6  此时6不存在于数组将此时的{7,1}记录
         3. numbers[i] = 9 则 res = 4  此时4不存在于数组将此时的{9,2}记录
         4. numbers[i] = 11 则 res = 2 此时2存在于数组所以hash.find(res) != hash.end()
            res=2 通过hash[res]计算出容器的index; i = 3!
        */
        if (hash.find(res) != hash.end()) {
            return{ hash[res] ,i};
        }

        hash[numbers[i]] = i; 
    }
    return{};
}

LeetCode运行结果如下：

最后给出一个相同的python版本，原理是想通的。

def twoSum(nums, target):
    dict = {}
    for i in range(len(nums)):
        if nums[i] in dict:
            return [dict[nums[i]], i]
        else:
            dict[target - nums[i]] = i

print(twoSum([2,7,9,11], 13))