題目:
給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個陣列中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target =
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
陣列至少兩個數值
第一反應是用兩個for迴圈,但是這是最簡單的思想。
然後再細想,可以在相加之前加上判斷,相加的兩個數不可以比target大。目前也就想到這裡。哈哈,這裡想錯了,沒有想到有負數出現的情況,應該是第二個數必須大於target-第一個數
#include <iostream> #include<vector> using namespace std; vector<int> twoSum(vector<int>& nums, int target) { vector<int> ans; int len = nums.size(); if (len <= 1) { cout << "error" << endl; return ans; } for (int i = 0; i<len; i++) { int diff = target - nums[i]; for (int j = i + 1; j<len; j++) { if (diff <= nums[j]) { if (nums[i] + nums[j] == target) { ans = { i, j }; return ans; } } } } return ans; } int main() { vector<int> nums = { -3, 4, 3, 90 }; vector<int> ans = twoSum(nums, 0); std::cout << ans[0] << ans[1] << std::endl; return 0; }
29 / 29 個通過測試用例
執行用時:432 ms
服了。。。。。
向高手們學習:發現他們用了map來提升速度。
注:
方法二:
vector<int> twoSum(vector<int>& nums, int target) { vector<int> ans; int len = nums.size(); if (len <= 1) { cout << "error" << endl; return ans; } unordered_map<int, int> map1; for (int n = 0; n<len; n++){ map1[nums[n]] = n;//key=陣列裡的數,value=序號 } for (int i = 0; i<len; i++) { int diff = target - nums[i]; if (map1.find(diff) != map1.end() && map1.find(diff)->second !=i) { ans = { i, map1.find(diff)->second }; return ans; } } return ans; }
執行用時:20 ms 舒服
方法三:
避免了判斷是否和自身相同
vector<int> twoSum(vector<int>& nums, int target) { vector<int> ans; int len = nums.size(); if (len <= 1) { cout << "error" << endl; return ans; } unordered_map<int, int> map1; for (int i = 0; i<len; map1[nums[i]] = i,i++) { int diff = target - nums[i]; if (map1.find(diff) == map1.end() ) continue; ans = { map1.find(diff)->second ,i}; } return ans; }
執行用時:20 ms
這道題複習了vector map hashmap的知識,要記住hashmap map vector的基本用法以及操作,還要區分什麼時間用什麼好