Leetcode-两数之和-C++

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

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

示例:

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

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

法一:暴力法

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

法二:哈希表(哈希大法好)

vector<int> twoSum(vector<int>& nums, int target) {
	vector<int> result;
	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) {
			result.push_back(i);
			result.push_back(tmpmap[target - nums[i]]);
			break;
		}
	}
	return result;
}

法三:进阶版的哈希表

vector<int> twoSum(vector<int>& nums, int target) {
	vector<int> result;
	map<int, int> tmpmap;//键值为nums的值,变量值为nums下标

	for (int i = 0; i < nums.size(); ++i) {
		if (tmpmap.count(nums[i]) != 0) {
			result.push_back(tmpmap[nums[i]]);
			result.push_back(i);
			break;
		}
		tmpmap[target - nums[i]] = i;
	}
	return result;
}
法三作为法二的优化版,虽然时间复杂度都是O(n),但法三将法二的两个循环合并了,只用执行一次循环即可。

猜你喜欢

转载自blog.csdn.net/cat1992/article/details/80372617