1. 程式人生 > >leetcode 01兩數之和

leetcode 01兩數之和

給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。

你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。

示例:

給定 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) {
        vector<int> twosum;
        for(int i=0;i<nums.size();i++){
            for(int j=i+1;j<nums.size();j++){
                if((nums[i]+nums[j])==target){
                    twosum.push_back(i);
                    twosum.push_back(j);
                }
            }
        }
    return twoSum;
    }
};

普通雜湊表的寫法:

vector<int> twoSum(vector<int>& nums, int target) {
      vector<int> twosum;
      map<int, int> tmpmap;//鍵值為nums的值,變數值為nums下標
        for (int i = 0; i < nums.size(); i++) {
          tmpmap[nums[i]] = i;
      }
      for (int i = 0; i < nums.size(); i++) {
             if (tmpmap.count(target - nums[i]) != 0 && tmpmap[target-nums[i]]!=i) {
                 twosum.push_back(i);
                 twosum.push_back(tmpmap[target - nums[i]]);
                 return twoSum;
            }
      }

 }  

優化之後的雜湊表:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> tmpmap;           //鍵值為num的值,變數值為num的下標
        vector<int> twosum;
        
        for(int i = 0; i < nums.size(); i++){
            if(tmpmap.count(nums[i]) != 0){
                twosum.push_back(tmpmap[nums[i]]);
                twosum.push_back(i);
                return twosum;
            }
            tmpmap[target - nums[i]] = i;
        }
    }
};