LeetCode #1 两数之和

LeetCode #1 两数之和

题目描述

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

示例:

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

所以返回 [0, 1]

传送门：LeetCode #1 两数之和

解法1 暴力枚举

708ms，7MB。

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

解法2 两遍哈希

20ms，8.4MB。
第一个for循环存储映射信息，第二for循环查找映射表，找到即返回。
需要注意的是，要确保同一个数不被用两次。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> mapNum;
        for(int i = 0; i<nums.size();i++){
            mapNum[nums[i]]=i;
        }
        
        for(int i = 0;i < nums.size(); i++){
            if(mapNum.find(target - nums[i]) != mapNum.end() && mapNum[target - nums[i]] != i){
                return {i,mapNum[target - nums[i]]};
            }
        }
        return {};
    }
    
};

解法3 一遍哈希

16ms，8.1MB。
for循环，边查表，边映射。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> mapNum;
        
        for(int i = 0;i < nums.size(); i++){
            if(mapNum.find(target - nums[i]) != mapNum.end()){
                return {i,mapNum[target - nums[i]]};
            }
            mapNum[nums[i]]=i;
        }
        return {};
    }
    
};