[LeetCode] Two Sum 解题报告

—— write for my baby, mua

[题目]

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

[中文翻译]

给定一个整型数组，给定一个值target，返回两个数字的下标，它们的和为target。

输入保证有且仅有一个解。

例子：

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

nums[0] + nums[1] = 2 + 7 = 9,
返回 [0, 1].

[解题思路]

O(NlogN)

先将数组k从小到大排序，设置头尾下标pL和pR，当nums[pL]+nums[pR]>target时，pR--，当nums[pL]+nums[pR]<target时，pL++，当nums[pL]+nums[pR]==target时，返回pL，pR。

这个思路是利用了有序的性质，减少了不必要的比较。对于O(N^2)的情景，对于nums[i]要和所有nums[j] (j>i)分别计算sum和target的关系。而在数组有序的条件下，假设当前左边下标为pL，找到了右边下标pR，满足nums[pL]+nums[pR]<target && nums[pL]+nums[pR+1]>target，那么对于nums[pL]而言，已经不需要比较nums[j] (j<pR)，因为nums[pL]与nums[j] (j<pR)相加的结果一定小于target，因此对于nums[pL]的比较已经是充分的，可以开始考虑nums[pL+1]。对于nums[pL+1]，已经不需要考虑nums[j] (j>=pR+1)，因为nums[pL+1]>nums[pL] && nums[pL]+nums[pR+1]>target，所以nums[pL+1]+nums[pR+1]>target，所以对于nums[pL+1]，只需从pR开始往前考虑。

O(N)

事实上，我们并不需要进行比较，对于nums[i]，我们只需要知道target-nums[i]在之前有没有出现过。而查找一个数在之前有没有出现过，可以有很多方法。一个效率比较稳定的方法是二叉搜索树（BST）/平衡树（AVL），每次插入或查找的效率为O(logN)。另一个方法是Hash，如果Hash足够好，可以达到O(1)的效率，那么总的时间复杂度为O(N)。由于题目要求给出下标，因此，在搜索树或者Hash中，需要同时记录nums对应的下标。

[C++代码]

class Solution {
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		vector<int> res;
		map<int, int> myMap;
		map<int, int>::iterator it;

		for (int i = 0; i < nums.size(); i++) {
			it = myMap.find(target - nums.at(i));
			if (it == myMap.end()) {
				if (myMap.find(nums.at(i)) == myMap.end())
					myMap.insert(map<int, int>::value_type(nums.at(i), i));
			}
			else {
				res.push_back(it->second);
				res.push_back(i);
				return res;
			}

		}

		return res;
	}
};