对于leetcode 885. Boats to Save People中采用multiset解法

题目的意思是,用船载人过河,一个船最多载两人,但不能超载,求最少需要几条船?

题目的入参是各个人体重的序列和超载额,要求输出为船的数量.

由于入参序列是vector,用排序+折半查找是可以解决问题的,但是由于人的数量在不断的减少,对于vector来说每次循环变换vector的size并不是一个好方法.

所以采用mutliset天然带排序的数据结构比较合适.

class Solution {
public:
    int numRescueBoats(vector<int>& p, int l) {
        multiset<int> ms;
        for(int i=0;i<p.size();i++)
        {
            ms.insert(p[i]);
        }
        int n=0;
        while(ms.size() >0)
        {
            auto max=ms.rbegin();
            n++;
            int left = l-*max;
            ms.erase(ms.find(*max));
            if(ms.empty())
                break;
            auto leftidx = ms.upper_bound(left);
            if(leftidx != ms.begin())
            {
                ms.erase(--leftidx);
            }
        }
        return n;
    }
};

其中,rbegin当前最重的那个人.

upper_bound是找到大于left的最后一个位置,这个位置的前面的元素必然是<=left的,那么删除前面的一个元素,是合适的人中最重的那个人.