題目

給定一個整數陣列 nums 和一個整數目標值 target，請你在該陣列中找出 和為目標值 的那 兩個 整數，並返回它們的陣列下標。

你可以假設每種輸入只會對應一個答案。但是，陣列中同一個元素不能使用兩遍。

你可以按任意順序返回答案。

示例 1：

輸入：nums = [2,7,11,15], target = 9
輸出：[0,1]
解釋：因為 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

輸入：nums = [3,2,4], target = 6
輸出：[1,2]

示例 3：

輸入：nums = [3,3], target = 6
輸出：[0,1]

提示：

2 <= nums.length <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只會存在一個有效答案

程式碼

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> a;
        vector<int> b(2,-1);
        // for(int i=0;i<nums.size();i++){
 

        //     a.insert(pair<int,int>(nums[i],i));
        // }
        for(int i=0;i<nums.size();i++){
            if(a.count(target-nums[i])>0){
                b[0]=i;
                b[1]=a[target-nums[i]];
                break;
            }
            a.insert(pair<int,int>(nums[
 
i],i));
        }
        return b;
    }
};

提交結果

總結

一開始我把nums中的數一次性先放進map裡再去逐一檢查，這樣是不行的，會出現：

輸入：
[3,2,4]
6
輸出：
[0,0]
預期：
[1,2]

因為將[3,2,4]一次性放進map裡，當i=0指向3時，馬上發現map中的3，符合3+3=6，返回的是[0,0]

所以改成了先檢查當前i指向的數是否在map裡有配對上的，如果有，直接返回對應下標；如果沒有，再將<nums[i],i>插入map中。再進行下一次迴圈。