题目
输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,则输出任意一对即可。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[2,7] 或者 [7,2]
示例 2:
输入:nums = [10,26,30,31,47,60], target = 40
输出:[10,30] 或者 [30,10]
限制:
- 1 <= nums.length <= 10^5
- 1 <= nums[i] <= 10^6
思路
遍历数组,查看target - n 是否在集合中,如果存在,则找到两个数,否则,将当前数n插入集合。
代码
时间复杂度:O(n)
空间复杂度:O(n)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
unordered_set<int> uset;
for (auto n : nums) {
if (uset.count(target - n) > 0) {
res.push_back(n);
res.push_back(target - n);
break;
}
uset.insert(n);
}
return res;
}
};