leetcode 01两数之和

Posted 2021-02-11 bejm

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

给定 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;
        }
    }
};