leetcode-1094. 拼车

最新推荐文章于 2025-02-12 15:23:49 发布

原创最新推荐文章于 2025-02-12 15:23:49 发布 · 344 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种用于判断顺风车是否能根据给定的乘客行程计划和车辆容量完成接送任务的算法。通过记录每个站点上下车的人数变化，确保在任何时刻车内人数不超过车辆容量，从而实现有效行程规划。

假设你是一位顺风车司机，车上最初有 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

#include <iostream>
#include <vector>
#include <string.h>

using namespace std;
bool carPooling(vector<vector<int>>& trips, int capacity) {
    int nums[1000];
    memset(nums, 0, sizeof(nums));
    long cnt =  0;
    for (int i = 0; i < trips.size(); i++) {
        nums[trips[i][1]] += trips[i][0]; // 上车人数
        nums[trips[i][2]] -= trips[i][0]; // 下车人数
    }

    for (int i = 0; i < 1000; i++) {
        cnt += nums[i];
        if (cnt > capacity) {
            return false;
        }
    }
    return true;
}

int main() {
    vector<vector<int>> trips = {{2,1,5},{3,3,7}};
    cout << carPooling(trips,  4) << endl;
}