1. 程式人生 > >第十六課:救生艇

第十六課:救生艇

問題描述

第 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

結果展示

在這裡插入圖片描述