leetcode 01兩數之和
給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9 因為 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
暴力法:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> twosum;
for(int i=0;i<nums.size();i++){
for(int j=i+1;j<nums.size();j++){
if((nums[i]+nums[j])==target){
twosum.push_back(i);
twosum.push_back(j);
}
}
}
return twoSum;
}
};
普通雜湊表的寫法:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> twosum;
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) {
twosum.push_back(i);
twosum.push_back(tmpmap[target - nums[i]]);
return twoSum;
}
}
}
優化之後的雜湊表:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> tmpmap; //鍵值為num的值,變數值為num的下標
vector<int> twosum;
for(int i = 0; i < nums.size(); i++){
if(tmpmap.count(nums[i]) != 0){
twosum.push_back(tmpmap[nums[i]]);
twosum.push_back(i);
return twosum;
}
tmpmap[target - nums[i]] = i;
}
}
};