LeetCode 1094. Car Pooling (拼车)

题目

链接

https://leetcode-cn.com/problems/car-pooling/

问题描述

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

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

必须接送的乘客数量；
乘客的上车地点；
以及乘客的下车地点。
这些给出的地点位置是从你的 初始 出发位置向前行驶到这些地点所需的距离（它们一定在你的行驶方向上）。

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

示例

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

提示

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

思路

差分数组的思路，设置differ数组存放每个站点和前一个站点的人数差，如果在第1站上车，第五站下车的话，那么第一站就比第0站多了n个人，2345没有区别，第六站的人就比第五站少了n个人。

借助差分数组计算完数量后，可以统计每站时有多少人，要求是每一站都小于容量。

复杂度分析

时间复杂度 O(max(m,n))
空间复杂度 O(n)

代码

Java

    public static boolean carPooling(int[][] trips, int capacity) {
        int[] differ = new int[1001];
        int[] ans = new int[1001];
        for (int[] trip : trips) {
            int num = trip[0];
            int in = trip[1];
            int out = trip[2] - 1;
            differ[in] += num;
            if (out + 1 < 1001) {
                differ[out + 1] -= num;
            }
        }
        ans[0] = differ[0];
        for (int i = 1; i < ans.length; i++) {
            ans[i] = ans[i - 1] + differ[i];
        }
        for (int i = 0; i < ans.length; i++) {
            if (capacity < ans[i]) {
                return false;
            }
        }
        return true;
    }