题目描述
输入一个数组和一个数字 s,在数组中查找两个数,使得它们的和正好是 s。
如果有多对数字的和等于 s,输出任意一对即可。
你可以认为每组输入中都至少含有一组满足条件的输出。
样例
输入:[1,2,3,4] , sum=7
输出:[3,4]
递增有序的数组 寻找两个数 ,这样的条件就宽泛了很多。
有序的数组可以考虑使用 二分查找,复杂度为O(nlogn)
还可以和leetcode 第一题两数之和解法类似,可以使用双指针和哈希的做法 复杂度O(n);
算法1
(二分) $O(logn)$
C++ 代码
class Solution {
public:
bool Check(const vector<int>& v, int idx, int find) {
return v[idx] >= find;
}
bool bSearch(const vector<int>& v, int l, int r, int find)
{
bool ret = false;
if (l < 0 || l >= v.size() || r < 0 || r >= v.size()) return ret;
while (l < r) {
int mid = l+r >> 1;
if (Check(v,mid,find)) r = mid;
else l = mid + 1;
}
return v[l] == find;
}
vector<int> twoSum(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); i++) {
int find = target - nums[i];
bool ret = false;
if (find > nums[i]) {
ret = bSearch(nums,i+1,nums.size()-1,find);
if (ret) {
return vector<int>{nums[i], find};
}
}
else if (find < nums[i]) {
ret = bSearch(nums, 0, i - 1, find);
if (ret) {
return vector<int>{nums[i], find};
}
}
else {
//说明查找的数字和nums[i]相等 直接查找该数字前后是否有该数即可
if (i - 1 >= 0 && nums[i - 1] == nums[i]) return vector<int>{find, find};
if (i + 1 < nums.size() && nums[i + 1] == nums[i]) return vector<int>{find, find};
}
}
return vector<int>();
}
};
算法2
(哈希) $O(n)$
C++ 代码
class Solution {
public:
unordered_set<int> ss;
vector<int> twoSum(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); i++) {
int find = target - nums[i];
if (find < 0) return vector<int>();
if (ss.count(find) != 0) return vector<int>{nums[i], find};
ss.insert(nums[i]);
}
return vector<int>();
}
};
算法3
(双指针) $O(n)$
C++ 代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int l = 0; int r = nums.size() - 1;
while (l < r) {
if (target == nums[l] + nums[r]) {
return vector<int>{nums[l], nums[r]};
}
else if (target > nums[l] + nums[r]) {
l++;
}
else if (target < nums[l] + nums[r]) {
r--;
}
}
return vector<int>();
}
};
这道题没说递增吧,,
剑指OFFER57 有提到递增。
如果没提的,增加个排序即可,不过时间复杂度就变成了nlogn
算法3加个排序:
sort(nums.begin(), nums.end());