题目链接

英文链接：https://leetcode.com/problems/car-pooling/

中文链接：https://leetcode-cn.com/problems/car-pooling/

题目详述

假设你是一位顺风车司机，车上最初有 capacity 个空座位可以用来载客。由于道路的限制，车只能向一个方向行驶（也就是说，不允许掉头或改变方向，你可以将其想象为一个向量）。

这儿有一份行程计划表 trips[][]，其中 trips[i] = [num_passengers, start_location, end_location] 包含了你的第 i 次行程信息：

必须接送的乘客数量；
乘客的上车地点；
以及乘客的下车地点。

这些给出的地点位置是从你的初始出发位置向前行驶到这些地点所需的距离（它们一定在你的行驶方向上）。

请你根据给出的行程计划表和车子的座位数，来判断你的车是否可以顺利完成接送所用乘客的任务（当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false）。

示例 1：

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

示例 2：

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

示例 3：

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

示例 4：

1 2	输入：trips = [[3,2,7],[3,7,9],[8,3,9]], capacity = 11 输出：true

提示：

你可以假设乘客会自觉遵守 “先下后上” 的良好素质
trips.length <= 1000
trips[i].length == 3
1 <= trips[i][0] <= 100
0 <= trips[i][1] < trips[i][2] <= 1000
1 <= capacity <= 100000

题目详解

本题直接解答比较复杂，可以采用统计学的方法进行加加减减操作，可以很容易解答本题。
因为 0 <= trips[i][1] < trips[i][2] <= 1000，可以开一个大小为 1001 的数组 cnt 来代表每个地点的人数。
遍历 trips，在上车点加上对应人数，在下车点减去对应人数。
最终数组 cnt 每个位置的前缀和就是对应地点上的乘客数量，判断是否满足条件就比较简单了。

public class LeetCode_01094 {

    public boolean carPooling(int[][] trips, int capacity) {
        int[] cnt = new int[1001];
        for (int[] trip : trips) {
            cnt[trip[1]] += trip[0];
            cnt[trip[2]] -= trip[0];
        }
        if (cnt[0] > capacity) return false;
        for (int i = 1; i < 1001; ++i) {
            cnt[i] += cnt[i - 1];
            if (cnt[i] > capacity) {
                return false;
            }
        }
        return true;
    }
}

如果去掉 0 <= trips[i][1] < trips[i][2] <= 1000 这个条件，就不能用数组来进行统计操作，需要用 Map，并且是 TreeMap，因为需要根据乘车顺序求前缀和。
上面是用加法判断是否大于 capacity，下面是用减法判断 capacity 是否小于 0。

public class LeetCode_01094 {

    public boolean carPooling(int[][] trips, int capacity) {
        Map<Integer, Integer> map = new TreeMap<>();
        for (int[] trip : trips) {
            map.put(trip[1], map.getOrDefault(trip[1], 0) + trip[0]);
            map.put(trip[2], map.getOrDefault(trip[2], 0) - trip[0]);
        }
        for (int v : map.values()) {
            capacity -= v;
            if (capacity < 0) {
                return false;
            }
        }
        return true;
    }
}