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;}
};