001-twosum

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
通过次数1,201,151提交次数2,452,456

 

// 该方法跑了两次，理论上一次可以完成

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

class Solution{
public:
    vector<int> twoSum(vector<int>& nums, int target){
        unordered_map<int, int>myhash;
        vector<int> out;
        for(int i = 0; i < nums.size(); i++){
            myhash[nums[i]] = i;
        }
        for(int i = 0; i < nums.size(); i++){
            if(myhash[target-nums[i]] && myhash[target-nums[i]] != i){ // 验证 nums[i] 和 target-nums[i]的和是否均在hashmap内部， 并验证的两个加数之一是自身
                out.push_back(i);
                out.push_back(myhash[target-nums[i]]);
                return out;
            }
        }
        return out;
    }
};


int main()
{
    Solution so;

    vector<int> test;
    int array[4] = {1,2,4,6};
    vector<int> testnum(array, array+4);
    test = so.twoSum(testnum, 5);
    for(vector<int>::iterator iter = test.begin(); iter != test.end(); iter++){
        cout << *iter << endl;
    }
 

 

解法二：一遍hash表

class Solution1{
public:
    vector<int> twoSum(vector<int> &nums, int target){
        vector<int> out;
        unordered_map<int, int> myhash;
        for(int i = 0 ; i < nums.size(); i++){
            int complement = target - nums[i];
            if(myhash.find(complement) == myhash.end()){
                myhash[nums[i]] = i;
            }else{
                out.push_back(i);
                out.push_back(myhash.find(complement)->second);
                return out;
            }
        }
        return out;
    }
};