【LeetCode】两数之和
题目:
在数组中找到 2 个数之和等于给定值的数字,结果返回 2 个数字在数组中的下标。要求时间复杂度为 O(n)。
解题分析:
作者:halfrost
链接:https://leetcode.cn/leetbook/read/leetcode-cookbook/5lu4og/
顺序扫描数组,对每一个元素,在 map 中找能组合给定值的另一半数字,如果找到了,直接返回 2 个数字的下标即可。如果找不到,就把这个数字存入 map 中,等待扫到“另一半”数字的时候,再取出来返回结果。
代码:
作者:JasonChiu
链接:https://leetcode.cn/leetbook/read/leetcode-cookbook/5lu4og/?discussion=789HT4
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> m;
for(int i = 0 ; i < nums.size(); ++i){
if( m.count( target - nums[i] ) == 0 ){
m[ nums[i] ] = i;
} else {
return { m[ target - nums[i] ], i};
}
}
return {};
}
};