LeetCode 刷题 [C++] 第[1]题. 两数之和

题目描述

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

示例:

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

解法一：粗暴解法

1、 两层for循环，最外层为0～n-2，里层为i~n-1；
2、 判断nums[i]+nums[j]==target是否为真，为真则返回两者坐标。
该方法时间复杂度为O(n^2)

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

运行结果：执行时间为56ms。

解法二：利用哈希表

将遍历后的元素值作为key：
1、每次循环时，先判断哈希表中是否包含target - nums[i]的值：
若包含，则返回对应的value和当前坐标；
若不包含，则将该元素值作为key，地址作为value，加入到哈希map中。
该方法时间复杂度为O(N)；

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        auto size = nums.size();
        unordered_map<int,int> value_map(size);
        vector<int> res;
        res.reserve(2);
        for(auto i=0;i<size;++i) {
            auto temp = target - nums[i];
            if(!value_map.count(temp)) value_map[nums[i]] = i;
            else { 
                res.emplace_back(value_map[temp]);
                res.emplace_back(i);
                break;
                }
        }
        return res;
    }
};

执行结果：执行时间为4ms。