LeetCode刷题系列 -- 1094. 拼车

车上最初有 capacity 个空座位。车 只能 向一个方向行驶（也就是说，不允许掉头或改变方向）

给定整数 capacity 和一个数组 trips , trip[i] = [numPassengersi, fromi, toi] 表示第 i 次旅行有 numPassengersi 乘客，接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。

当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false。

示例 1：

输入：trips = [[2,1,5],[3,3,7]], capacity = 4

输出：false

示例 2：

输入：trips = [[2,1,5],[3,3,7]], capacity = 5

输出：true

提示：

• 1 <= trips.length <= 1000

• trips[i].length == 3

• 1 <= numPassengersi <= 100

• 0 <= fromi < toi <= 1000

• 1 <= capacity <= 105

1094. 拼车 - 力扣（Leetcode）

思路：小而美的算法技巧：差分数组 :: labuladong的算法小抄

本题也是可以利用 差分数组来求解

c++

class Solution {
public:
    vector<int> diff;
    
    void increment(vector<int>& diff, int i, int j, int val){
        diff[i] += val;
        if(j < diff.size()) {
            diff[j] -= val;
        }
    }

    bool carPooling(vector<vector<int>>& trips, int capacity) {
        diff = vector<int>(1001, 0);

        vector<int> result(1001, 0);

        int maxLen = 0;
        for(auto& trip:trips) {
            int numPassengers = trip[0];
            int from = trip[1];
            int to = trip[2];
            maxLen = max(maxLen, to);
            increment(diff, from, to, numPassengers);
        }

        result[0] = diff[0];
        for(int i=1; i<=maxLen; i++) {
            result[i] = result[i-1] + diff[i];
        }
        bool flag = true;
        for(int i=0; i<=maxLen; i++) {
             if(result[i] > capacity) {
                 flag =  false;
             }
        }


        return flag;
    }
};