Leetcode_两数之和

题目

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

示例：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]

暴力解法

两个循环遍历数组找到和为target的两个数，复杂度为O(n²)。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> ret;
        for(int i = 0; i < nums.size() - 1; i++){
            for(int j = i + 1; j < nums.size(); j++){
                if(nums[i] + nums[j] == target){
                    ret.push_back(i);
                    ret.push_back(j);
                    return ret;
                }
            }
        }
    return ret;
    }
};

其实这里可以直接返回标号，避免调用push_back()函数，如下：

return {i, j};

哈希表解法

参考leetcode题解，复杂度为O(1)。

class Solution{
public:
    vector<int> twoSum(vector<int>& nums, int target){
		vector<int> ret(2, -1);
		map<int, int> num;
		for(int i = 0; i < nums.size(); i++){
		    if(num.count(target - nums[i]) > 0){
				ret[0] = i;
				ret[1] = num[target - nums[i]];
				return ret;
		    }
		    num[nums[i]] = i;	//key为数值，value为下标
		}
	    return ret;
    }
};

思路：对每一个nums中的元素值nums[i]查找该数组中是否含有解target-nums[i]。
将nums中的元素值和索引值存入map结构中，遍历nums并在map中查找是否含有解（利用count()函数）。
因为只需要一个解所以nums中的每一个元素都可以放入map中，不需要考虑碰撞问题。

数据结构map的使用方法参考