题目的意思是,用船载人过河,一个船最多载两人,但不能超载,求最少需要几条船?
题目的入参是各个人体重的序列和超载额,要求输出为船的数量.
由于入参序列是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的,那么删除前面的一个元素,是合适的人中最重的那个人.