题目描述
给定两个大小相等的数组 A
和 B
,A 相对于 B 的优势可以用满足 A[i] > B[i]
的索引 i
的数目来描述。
返回 A
的任意排列,使其相对于 B
的优势最大化。
示例 1:
输入:A = [2,7,11,15], B = [1,10,4,11] 输出:[2,11,7,15]
示例 2:
输入:A = [12,24,8,32], B = [13,25,32,11] 输出:[24,32,8,12]
提示:
1 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9
解题思路
贪心:对于B中的每一个元素B[i] {i = 0,....len-1},在A中找第一个大于B[i]的A[j],如果不存在则取最小的
vector<int> advantageCount(vector<int>& A, vector<int>& B) {
vector<int> ans;
vector<int>::iterator it;
sort(A.begin(),A.end());
int len = B.size();
for(int i=0;i<len;i++){
it = upper_bound(A.begin(),A.end(),B[i]);
if(it == A.end()) it = A.begin();
ans.push_back(*it);
A.erase(it);
}
return ans;
}