Two Sum问题

要求

给定一个数组和一个目标数，在数组里查找两个数，这两个数之和等于目标数，返回这两个数的下标。
例如：nums = [2,7,11,15],target = 9
return [0,1]

做法

遍历数组，利用hash表存储这个数组，键为数组元素，值为数组下标。遍历过程中，如果hash表内存在(target-当前元素),则找到sum为target的两个数。
这种做法的时间复杂度为O(n)。

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

class Solution {
public:
     vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        unordered_map<int, int> hash_map;
        for (int i=0; i<nums.size(); i++) {
            if (hash_map.count(nums[i])==0) {
                hash_map.insert(unordered_map<int, int>::value_type(nums[i],i));

            }
            if (hash_map.count(target-nums[i])==1) {
                int index = hash_map.find(target-nums[i])->second;
                if (i<index) {
                    result.push_back(i);
                    result.push_back(index);
                }else if(i>index){
                    result.push_back(index);
                    result.push_back(i);
                }
            }
        }
        return result;
    }
};

Comments