[Greedy] leetcode 870 Advantage Shuffle

最新推荐文章于 2024-01-14 18:00:38 发布

ayzejx8547408

最新推荐文章于 2024-01-14 18:00:38 发布

阅读量101

点赞数

原文链接：http://www.cnblogs.com/fish1996/p/11370967.html

版权

problem: https://leetcode.com/problems/advantage-shuffle/

In ordered to maximize the advantage of array A with respect to B, we had better choose the smallest element in array A that is larger than the element in B. After each selection, we erase the data we choose in A to avoid repetition.

class Solution {
public:
    vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        vector<int> res;
        multiset<int> datas(A.begin(), A.end());
        for(int i = 0;i < B.size(); i++)
        {
            // find the smallest number in array A that is larger than the element in array B
            auto it = datas.upper_bound(B[i]);
            
            if(it != datas.end())
            {
                res.push_back(*it);
                datas.erase(it);
            }
            else
            {
                res.push_back(*datas.begin());
                datas.erase(datas.begin());
            }
        }
        
        return res;
    }
};