c_cpp 1.两个总和

Posted 2021-05-14

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res(2);
        unordered_map<int, int> mp;
        
        for (int i = 0; i < nums.size(); i++) {
            if (mp.count(target - nums[i])) {
                res[0] = mp[target - nums[i]];
                res[1] = i;
                return res;
            }
            else {
                mp[nums[i]] = i;
            }
        }
        
        return res;
    }
};