Leetcode-1094. 拼车

链接

1094. 拼车

题目

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

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

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

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

示例

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

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

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

示例 4：
输入：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

思路

参考力扣精选评论

由于map容器本身有序， 我们可以把上车地点和上车乘客人数，以及下车地点和下车人数都存放map中， 然后遍历整个map，对数据进行累加，变可得到实时乘客人数。

C++ Code

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        map<int, int> m;
        for(auto & t: trips){
            m[t[1]] += t[0];  // 上车
            m[t[2]] -= t[0];  // 下车
        }
        int passengers = 0;
        for(auto it: m){
            passengers += it.second;//实时计算当前人数
            if(passengers > capacity) return false;
        }
        return true;
    }
};

此外，由于本题题目已经说了trips.length <= 1000，因此，可以直接定义一个大小为1001的数组存放每个路段上下车的人数，思路和上述代码一致。

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        vector<int> road(1001, 0);
        for(int i=0;i<trips.size();i++)
        {
            road[trips[i][1]]+=trips[i][0];
            road[trips[i][2]]-=trips[i][0];
        }
        int count=0;
        for(int i=0;i<road.size();i++)
        {
            count+=road[i];
            if(count>capacity) return false;
        }
        return true;

    }
};