天池在线编程最佳利用率（二分查找 + 哈希）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给两个排序的数组。
从两个数组中各取取一个数，这两个数之和需要小于或等于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]

https://tianchi.aliyun.com/oj/338592228998370026/357527484118536803

2. 解题

遍历 B 的元素，记录每个值第一次出现的位置
遍历 A，在 B 中查找小于等于 K-A[i] 的最大的数
时间复杂度 $O(n_1\log n_2)$

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 hereif(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;}elsel = mid+1;}}if(pos != -1){sum = A[i]+B[pos];if(sum > maxsum){maxsum = sum;ans[0] = i;ans[1] = m[B[pos]];}}elsebreak;}return ans;}
};