力扣OJ 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 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/advantage-shuffle
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

思路：

田忌赛马

对于B的任意序遍历，每次找A中比B[i]大的最小的A[j]，如果没有就选最小的A[k]

代码：

class Solution {
public:
    vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        vector<int>ans;
        multiset<int>mset;
        multiset<int>::iterator it;
        ans.clear();
        mset.clear();
        for(int i=0;i<A.size();i++){
            mset.insert(A[i]);
        }
        for(int i=0;i<B.size();i++){
            it=mset.upper_bound(B[i]); //找比B[i]小的最小A[?]
            if(it==mset.end()){
                it=mset.begin(); //如果没有就取A[]中最小数
            }
            ans.insert(ans.end(),*it);
            mset.erase(it);
        }
        return ans;
    }
};