第十六課:救生艇
阿新 • • 發佈:2018-12-30
問題描述
第 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
解決方法
方法:貪心(雙指標)
思路
如果最重的人可以與最輕的人共用一艘船,那麼就這樣安排。否則,最重的人無法與任何人配對,那麼他們將自己獨自乘一艘船。
這麼做的原因是,如果最輕的人可以與任何人配對,那麼他們也可以與最重的人配對。
演算法
令 people[i] 指向當前最輕的人,而 people[j] 指向最重的那位。
然後,如上所述,如果最重的人可以與最輕的人共用一條船(即 people[j] + people[i] <= limit),那麼就這樣做;否則,最重的人自己獨自坐在船上。
c++語言:
class Solution { public: int numRescueBoats(vector<int>& people, int limit) { sort(people.begin(), people.end()); int i = 0, j = people.size() - 1; int ans = 0; while (i <= j) { ans++; if (people[i] + people[j] <= limit) i++; j--; } return ans; } };
Python語言:
class Solution(object):
def numRescueBoats(self, people, limit):
people.sort()
i, j = 0, len(people) - 1
ans = 0
while i <= j:
ans += 1
if people[i] + people[j] <= limit:
i += 1
j -= 1
return ans