LeetCode1094.拼车

1.题目

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

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

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

2.题解

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        int to_max=0;
        for(auto x: trips){
            to_max=max(to_max,x[2]);
        }
        vector<int> diff(to_max+1);
        for(auto x:trips){
            diff[x[1]]+=x[0];
            diff[x[2]]-=x[0];
        }
        int cnt=0;
        for(int i=0;i<=to_max;i++){
            cnt+=diff[i];
            if(cnt>capacity){
                return false;
            }
        }
        return true;
    }
};

解题基本思路，维护区间上加上某数以及区间并集上的最大值，这里使用，维护区间和的时候可以考虑前缀和和差分，差分是前缀和的逆运算，可以记录每个位置的变化量。记录差分之后再用O(n)的时间来计算前缀和，可以维护区间值。