题目来源:https://leetcode-cn.com/problems/boats-to-save-people/
大致题意:
给出一个乘客的重量数组和一个船的限重,其中单个乘客的重量不会超过限重。
每辆船最多载两个人,求出乘客全部载走需要的船数量。
思路
贪心
要想让船的数量尽可能的小,就需要让每次船载人的重量尽可能的大,又不会超过限重。
而一次次枚举两个最合适的重量的算法复杂度为 O(n^3),显然不合适。
排序 + 双指针
于是可以先对重量数组进行排序,每次挑出一个当前最重的和最轻的上船。若重量和不超过限重,则都能上船;若超过,则只上重的那个人。重复上述过程,直至所有人都上船。
这样求得的数量即为所求答案。证明:
- 假定现在有排序好(升序)的重量数组 a1, a2, a3 … a8, a9,假设每次选最重和最轻的不是最佳选择,那就是次重和最轻合适,让最重和次轻一块
- 按照刚刚的规则,应该选 a9 + a2 a8 + a1 上船,这样如果都不超重,那么同理 a9 + a1(< a2), a8(< a9) + a2 显然更加合适;而按照之前的规则 a9 + a1, a8 + a2 如果超重,那么 a9 + a2(> a1) 显然更超重,只能选 a9 单独一辆船合适。
代码:
public int numRescueBoats(int[] people, int limit) {
int num = 0;
// 排序
Arrays.sort(people);
// 双指针
int light = 0;
int heavy = people.length - 1;
while (light <= heavy) {
// 指针相遇,只剩一个人
if (light == heavy) {
num++;
break;
}
// 未超重,都上船
if (people[light] + people[heavy] <= limit) {
light++;
heavy--;
num++;
}
else {
heavy--;
num++;
}
}
return num;
}