当前位置：首页 > article >正文

LC.1094. 拼车（差分）

article 2025/2/25 3:22:37

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

差分

class Solution {
    public boolean carPooling(int[][] trips, int capacity) {
        int[] p = new int[1005];
        int x = capacity, i = 0;
        Arrays.sort(trips, (a, b) -> a[1] - b[1]);
        for(int[] t : trips){
            int passenger = t[0], from = t[1], to = t[2];
            
            while(i <= from)
                x += p[i++];
            if(x < passenger) return false;
            x -= passenger;
            // 需要注意的是在to就下车了，所以这里加在p[to]
            p[to] += passenger;
        }
        return true;
    }
}