870. Advantage Shuffle

问题：

给定两个size相同的数组A，B

对A进行排列，使得同位 i 上对数，A[i]>B[i]，求使得满足这样条件元素最多的A的排列。

Example 1:
Input: A = [2,7,11,15], B = [1,10,4,11]
Output: [2,11,7,15]

Example 2:
Input: A = [12,24,8,32], B = [13,25,32,11]
Output: [24,32,8,12]
 

Note:
1 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9

　　

解法：贪婪算法

思想：对A进行从小到大排序，

遍历B，对B的每一个元素b，在A中从小到大找A[i]>b最小的A[i]，

若不存在这样的A[i]，则用当前A中最小的A[0]去对应b。

代码参考：

class Solution {
public:
    vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        vector<int> res;
        int i=0;
        sort(A.begin(), A.end());
        for(int b:B){
            int size=A.size();
            for(i=0; i<A.size(); i++){
                if(A[i]>b){
                    res.push_back(A[i]);
                    A.erase(A.begin()+i);
                    break;
                }
            }
            if(A.size()>0 && i==size){
                res.push_back(A[0]);
                A.erase(A.begin());
            }
        }
        return res;
    }
};

♻️优化，使用C++的multiset，（保存排序A）

使用upper_bound(b)二分查找来找到第一个大于b的数值。

代码参考：

static auto x = []() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    return NULL;
}();

class Solution {
public:
    vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        multiset<int> s(A.begin(), A.end());
        vector<int> res;
        for(int b:B){
            auto p=s.upper_bound(b);
            if(p==s.end()) p=s.begin();
            res.push_back(*p);
            s.erase(p);
        }
        return res;
    }
};