天天看點

LeetCode_1. 兩數之和

題目描述:

給定一個整數數組 nums 和一個目标值 target,請你在該數組中找出和為目标值的那 兩個 整數,并傳回他們的數組下标。

你可以假設每種輸入隻會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。

示例:

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

因為 nums[0] + nums[1] = 2 + 7 = 9
是以傳回 [0, 1]      
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> m;
        set<int> s;
        for(int i=0;i<nums.size();i++){
            if(s.find(target-nums[i])!=s.end()){
                return {m[target-nums[i]],i};
            }
            else{
                s.insert(nums[i]);
                m[nums[i]]=i;
            }
        }
        return {};
    }
};