【数据结构-一维差分】力扣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

差分

class Slution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        int maxNum = INT_MIN;
        for(auto trip : trips){
            maxNum = max(maxNum,trip[2]);
        }
        vector<int> diff(maxNum+1);
        
        for(auto trip : trips){
            diff[trip[1]]+=trip[0];
            diff[trip[2]]-=trip[0];
        }

        int s = 0, ans = 0;
        for(int d : diff){
            s += d;
            if(capacity < s){
                return false;
            }   
        }
        return true;
    }
};

这题运用差分的思想，我们首先定义一个差分数组diff来记录差分，他的大小是最远的站数+1。

然后遍历每个trip，由于diff的目的是记录每一次trip的人数的差分，所以当某一次trip的人上车，那么diff就diff[trip[1]]+=trip[0];，当这次trip到终点的时候，也就是下车了，所以说终点是一个开区间，于是diff[trip[2]]-=trip[0]。

然后我们定义一个s来记录，假设车容量没有限制的话，车上同时会有多少人，当发现车上同时拥有的人数大于capacity容量的时候，就说明无法在所有给定的行程中接送所有乘客，如果车上人数始终在capacity内，那么说明可以按需求接送所有乘客。