题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
一刷 暴力解法
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
for(int i=0;i<nums.size()-1;i++){
for(int j=i+1;j<nums.size();j++){
if(nums[i]+nums[j]==target){
ans.push_back(i);
ans.push_back(j);
}
}
}
return ans;
}
};
map解法
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> codes;
vector<int> ans;
for(int i=0;i<nums.size();i++){
auto iter = codes.find(target-nums[i]);
if(iter!=codes.end()){
ans.push_back(iter->second);
ans.push_back(i);
}
codes.insert(pair<int,int>(nums[i],i));
}
return ans;
}
};
总结
- map的插入用pair,也可以事先申请pair再插入
- map中find函数的使用,利用迭代器配合编程