leetcode刷题笔记——array
001 两数之和
题目链接
思路1:
直接暴力枚举,即使用双重循环,逐个判断nums[i]+nums[j]与target的大小关系。时间复杂度为o(n^2)
思路2:
正如我们求x+y+z=C(C>0)的正整数解一样,我们可以不全部枚举x,y,z,只枚举x,y,最后一个根据
C-x-y求得。这里也是一样的,可以只枚举其中一个数,存下枚举过的数的下标和数值,之后遇到的每个数都先在表里面判断是否有满足条件的数值存在,如果有,则记录答案并返回,如果没有,就将当前的数和位置加到表里。
代码如下:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int>m;
vector<int>ans;
int n=nums.size();
for(int i=0;i<n;i++){
if(m.find(nums[i])!=m.end()){
ans.push_back(i);
ans.push_back(m[nums[i]]);
return ans;
}
m[target-nums[i]]=i;
}
return ans;
}
遇到的问题:
只想到用unordered_set记录下标,没想到用map同时记录下标和数值以至于卡住了。
或者一定要用set的话,也可以存pair<int,int>
类型,保存键值对。