Leetcode 870：优势洗牌

题目描述

给定两个大小相等的数组 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. 1 <= A.length = B.length <= 10000
2. 0 <= A[i] <= 10^9
3. 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;
    }