1. 题目
给你一个整数数组 A 和一个整数 K,请在该数组中找出两个元素,使它们的和小于 K 但尽可能地接近 K,返回这两个元素的和。
如不存在这样的两个元素,请返回 -1。
示例 1:
输入:A = [34,23,1,24,75,33,54,8], K = 60
输出:58
解释:
34 和 24 相加得到 58,58 小于 60,满足题意。
示例 2:
输入:A = [10,20,30], K = 15
输出:-1
解释:
我们无法找到和小于 15 的两个元素。
提示:
1 <= A.length <= 100
1 <= A[i] <= 1000
1 <= K <= 2000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
2.1 暴力
class Solution {
public:
int twoSumLessThanK(vector<int>& A, int K) {
int i, j, n = A.size(), sum, mindiff = INT_MAX;
for(i = 0; i < n-1; ++i)
{
for(j = i + 1; j < n; ++j)
{
if(A[i]+A[j] < K && K-A[i]-A[j] < mindiff)
{
mindiff = K-A[i]-A[j];
sum = A[i]+A[j];
}
}
}
return mindiff==INT_MAX ? -1 : sum;
}
};
4 ms 9.2 MB
2.2 二分查找
- 排序
- 固定一个数,二分查找另一个数
class Solution {
public:
int twoSumLessThanK(vector<int>& A, int K) {
sort(A.begin(), A.end());
int sum, mindiff = INT_MAX, l, r, mid, t;
for(int i = 0; i < A.size(); ++i)
{
l = 0, r = A.size()-1;
t = K-A[i];//找到小于t的最大数
while(l <= r)
{
mid = l+((r-l)>>1);
if(A[mid] >= t)
r = mid-1;
else
{
if(mid==A.size()-1 || A[mid+1] >= t)
{ //找到了
if(mid != i && K-A[mid]-A[i] < mindiff)
{ //不是同一个数,且更接近
mindiff = K-A[mid]-A[i];
sum = A[mid]+A[i];
}
break;
}
else
l = mid+1;
}
}
}
return mindiff==INT_MAX ? -1 : sum;
}
};
4 ms 9.1 MB
长按或扫码关注我的公众号,一起加油、一起学习进步!