LeetCode 881. 救生艇
题目描述
第 i 个人的体重为 people[i],每艘船可以承载的最大重量为 limit。
每艘船最多可同时载两人,但条件是这些人的重量之和最多为 limit。
返回载到每一个人所需的最小船数。(保证每个人都能被船载)。
示例 1:
输入:people = [1,2], limit = 3
输出:1
解释:1 艘船载 (1, 2)
示例 2:
输入:people = [3,2,2,1], limit = 3
输出:3
解释:3 艘船分别载 (1, 2), (2) 和 (3)
示例 3:
输入:people = [3,5,3,4], limit = 5
输出:4
解释:4 艘船分别载 (3), (3), (4), (5)
提示:
- 1 <= people.length <= 50000
- 1 <= people[i] <= limit <= 30000
解题思路
题目中已经假设了每个人都不会超过船的承重,并且每条船限载两人,所以只可能有两种情况:一个人一条船、两个人一条船。
对于瘦子来讲,可能和很多人拼船都是可以的,不会影响结果;但是对于胖子来讲,如果能和自己拼船的瘦子都去和别人拼船了,这时候他就只能选择自己一条船了。
如何避免这种情况呢?
考察一种经典场景:假设船的载重为w,四个人体重分别为 0.3w、0.4w、0.6w、0.7w,现在两个瘦子拼船了,胖子只能各自一条船,总共用了三条船。实际上,我们可以通过调整,让胖子和瘦子拼船,两条船就够了。
分配时,可以考虑让最胖的和最瘦的同乘一条船,如果这样都超重,那么此时胖子自己一条船就无话可说了。
实际上,这就是贪心策略。可以反证其有效性。
参考代码
class Solution {
public:
int numRescueBoats(vector<int>& people, int limit) {
if (people.empty()) return 0;
sort(people.begin(), people.end());
if (people[0] > limit) return -1;
int cnt = 0;
int i = 0;
int j = people.size() - 1;
while(i < j) {
if (people[i] + people[j] > limit) {
cnt++;
j--;
} else {
cnt++;
i++;
j--;
}
}
if (i == j) cnt++;
return cnt;
}
};