1. 程式人生 > >1.無序陣列中找出兩個數使其和等於給定值

1.無序陣列中找出兩個數使其和等於給定值

碰到這種類似題目可以根據條件不同寫出不同時間複雜度的程式碼:

1.最暴力的遍歷,時間複雜度為O(n^2)

2.一般情況下先排序,再從兩邊向中間搜尋結果,時間複雜度為O(nlogn + n)

int i = 0, j = numbers.size() - 1;
while (i < j)
{
	if (numbers[i] + numbers[j] == target) return true;
	else if (numbers[i] + numbers[j] < target) ++i;
	else --j;
}

3.如果陣列所有數都是唯一的,可以用unordered_map,時間複雜度為O(n),空間複雜度為O(n)

本題採用的是第三種方法accept,程式碼如下:

#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
class Solution {
public:	
	vector<int> twoSum(vector<int> &numbers, int target) {
		vector<int> result;
		unordered_map<int, int> NumMap;
		for (int i = 0; i < numbers.size(); ++i)
		{
			if (NumMap[target - numbers[i]])
			{
				result.push_back(NumMap[target - numbers[i]]);
				result.push_back(i + 1);
				break;
			}
			NumMap[numbers[i]] = i + 1;
		}
		return result;
	}
};
int main()
{
	int n, target;
	Solution so;
	while (cin >> n >> target)
	{
		vector<int> numbers(n);
		for (int i = 0; i < n; ++i) cin >> numbers[i];
		vector<int> res = so.twoSum(numbers, target);
		for (int i = 0; i < res.size(); ++i) cout << res[i] << " ";
		cout << endl;
	}
	return 0;
}