【算法百题之二十八】两数之和

【算法百题之二十八】两数之和

 

    大家好，我是Lampard~~

    很高兴又能和大家见面了，接下来准备系列更新的是算法题，一日一练，早日升仙！

 

    今天的问题是：

    

     

思路：

一开始的时候，我就直接想到暴力两个循环解决这个问题，这样子做的时间复杂度是o(nlogn)

for(int i = 0; i < nums.size(); i++){
    for(int j = i + 1; j < nums.size(); j++){
        if(nums[i] + nums[j] == target){
            // 输出结果
        }
    }
}

  结果超时，然后我想怎么把这个过程优化，一开始我想到如果有很多重复的元素，如果之前出现过这个元素，那么我们其实没有必要进入第二重循环

map<int, int> value2Index;
for(int i = 0; i < nums.size(); i++){
    if(value2Index.find(nums[i]) != value2Index.end()){
        continue;
    }
    value2Index[nums[i]] = i;
    for(int j = i + 1; j < nums.size(); j++){
        if(nums[i] + nums[j] == target){
            // 输出结果
        }
    }
}

 结果还是超时，这时候我突然想到，我既然已经想到用map键值对来进行优化，为什么我不直接使用map来进行判断呢？我们可以判断（target - nums【i】）的值是否出现，如果出现过，那么我们把该值和i一起输出就是答案，这样子做的话时间复杂度可以控制在o（n）

#include <iostream>
#include <vector>
#include <map>
using namespace std;

void printVector(vector<int> vectorTest)
{
	for (vector<int>::iterator it = vectorTest.begin(); it != vectorTest.end(); it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

vector<int> twoSum(vector<int>& nums, int target) {
	vector<int> result(2);
	map<int, int> value2Index;
	for (int i = 0; i < nums.size(); i++) {
		int key = target - nums[i];
		// 我们还可以通过value2Index.count(key) == 0 来判断是否存在key值
		if (value2Index.find(key) != value2Index.end()) {
			result[0] = i;
			result[1] = value2Index[target - nums[i]];
			return result;
		}
		value2Index[nums[i]] = i;
	}
	return result;
}

int main()
{
	vector<int> test = { 2, 7, 11, 15 };
	vector<int> result = twoSum(test, 9);
	printVector(result);
}

 测试函数以及测试结果：

 

   OK，今天的博客就到这里，谢谢大家！！！