LeetCode 1. 兩數之和

ln2037發表於2020-12-16

給定一個整數陣列 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>mp;vector<int>v;
        for(int i = 0; i < nums.size(); i++) {
            int k = target - nums[i];
            if(mp[k] != 0) {
                v.push_back(mp[k] - 1);
                v.push_back(i);
                return v;
            }
            mp[nums[i]] = i + 1;
        }
        return v;
    }
};

相關文章