Leecode1099号问题:
给你一个整数数组 A 和一个整数 K,请在该数组中找出两个元素,使它们的和小于 K 但尽可能地接近 K,返回这两个元素的和。
如不存在这样的两个元素,请返回 -1。
示例 1:
输入:A = [34,23,1,24,75,33,54,8], K = 60
输出:58
解释:
34 和 24 相加得到 58,58 小于 60,满足题意。
1
2
3
4
示例 2:
输入:A = [10,20,30], K = 15
输出:-1
解释:
我们无法找到和小于 15 的两个元素。
1
2
3
4
提示:
1 <= A.length <= 100
1 <= A[i] <= 1000
1 <= K <= 2000
解题思路
由于给定的数据两不打,所以我们不难想到暴力解法,直接枚举两个数,选择两数和小于K
的最大值即可。
显然这是一个两数和的问题,那么我们可以使用对撞指针两做,类似问题
Leetcode 167:两数之和 II - 输入有序数组
想用两数和的算法,我们首先对输入的A排序,然后建立左右指针l和r。我们判断左右指针所指向的两数之和tmp和K的关系,如果tmp>=K,那么我们r--,否则的话l++,并且通知记录我们的最终结果即可。
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
class Solution
{
public:
int twoSumLessThanK(vector<int>& A,int K)
{
int res = -1;
int len = A.size();
sort(A.begin(),A.end());
int l = 0;
int r = len-1;
while(l < r)
{
int tmp = A[l] + A[r];
if(tmp >= K)
{
r--;
}
else
{
res = max(tmp,res);
l++;
}
}
return res;
}
};
int main()
{
Solution solve;
int arr[] = {34,23,1,24,75,33,54,8};
int len = sizeof(arr)/sizeof(arr[0]);
vector<int> A(arr,arr+len);
int res = solve.twoSumLessThanK(A,60);
for(vector<int>::iterator it = A.begin();it != A.end();it++)
{
cout << *it << " ";
}
cout << endl;
cout << res << endl;
return 0;
}