给两个排序的数组。 从两个数组中各取取一个数,这两个数之和需要小于或等于k, 需要找到两数之和最大的索引组合。 返回一对包含两个列表的索引。 如果有多个两数之和相等的索引答案,你应该选择第一个数组索引最小的索引对。 在此前提下,你应该选择第二个数组索引最小的索引对。
1)两数的总和<= k 2)总和是最大的 3)两个数组索引都尽量最小
如果你无法找到答案,你应返回一个空列表[]。 You can assume that the numbers in arrays are all positive integer or zero.
示例1
输入:
A = [1, 4, 6, 9], B = [1, 2, 3, 4], K = 9
输出:
[2, 2]
示例2:
输入:
A = [1, 4, 6, 8], B = [1, 2, 3, 5], K = 12
输出:
[2, 3]
K-A[i]
的最大的数class Solution {
public:
/**
* @param A: an integer sorted array
* @param B: an integer sorted array
* @param K: an integer
* @return: return a pair of index
*/
vector<int> optimalUtilization(vector<int> &A, vector<int> &B, int K) {
// write your code here
if(A.empty() || B.empty() || A[0]+B[0] > K) return {};
unordered_map<int,int> m;
for(int i = 0; i < B.size(); ++i)
if(m.find(B[i]) == m.end())
m[B[i]] = i;//每个数最早出现的位置
int i = 0, j = B.size()-1, n1 = A.size(), n2 = B.size(), sum, maxsum=INT_MIN;
vector<int> ans(2);
for(int i = 0; i < n1; ++i)
{
int t = K-A[i];
int l = 0, r = n2-1, pos=-1;
while(l <= r) // 二分查找
{
int mid = l+((r-l)>>1);
if(B[mid] > t)
r = mid-1;
else
{
if(mid==n2-1 || B[mid+1] > t)
{
pos = mid;
break;
}
else
l = mid+1;
}
}
if(pos != -1)
{
sum = A[i]+B[pos];
if(sum > maxsum)
{
maxsum = sum;
ans[0] = i;
ans[1] = m[B[pos]];
}
}
else
break;
}
return ans;
}
};
50ms C++