1.無序陣列中找出兩個數使其和等於給定值
阿新 • • 發佈:2018-12-30
碰到這種類似題目可以根據條件不同寫出不同時間複雜度的程式碼:
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; }