Leetcode 1094. 拼车

Leetcode 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

思路

• 题目中也进行了提示， 看成一维向量， 所以我们直接用map记录每个上下车地点的人员变动情况
• 由于在一个位置可能既有上车的人， 也有下车的人， 所以map的操作全都设置成+=
• 最后的判断条件就是 —— 当前的车上人数<=capacity即可

代码

class Solution {
public:
    bool carPooling(vector<vector<int>>& trips, int capacity) {
        map<int, int> mp;
        int size = 0;

        for (auto& trip: trips) {
            mp[trip[1]] += trip[0];
            mp[trip[2]] += -trip[0];
        }

        for (auto& data:mp) {
            size += data.second;

            if (size > capacity) return false;
        }

        return true;
    }
};