LeetCode 每日一题 Day1_代码007(未授权)

本文介绍: 给定整数 capacity 和一个数组 trips , t rip[i] = [numPa sseng e rs i, fro m i, t o i] 表示第 i 次旅行有 numPa sseng e rs i 乘客，接他们和放他们的位置分别是 fr o mi 和 toi。当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false。输入：trips = [[2,1,5],[3,3,7]], cap a city = 4。输入：trips = [[2,1,5],[3,3,7]], cap a city = 5。

车上最初有 cap a city 个空座位。车只能向一个方向行驶（也就是说，不允许掉头或改变方向）

给定整数 c apacity 和一个数组 trips , tr ip[i] = [numPasseng e rsi, fromi, toi] 表示第 i 次旅行有 numPasseng e rsi 乘客，接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。

当且仅当你可以在所有给定的行程中接送所有乘客时，返回 true，否则请返回 false。

示例 1：

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

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

#include <iostream&gt;
#include <vector&gt;

bool carPooling(std::vector<std::vector<int&gt;>&amp; trips, int capacity) {
    const int maxLocation = 1001; // 最大位置数，根据题目要求设定

    // 创建差分数组，初始化为0
    std::vector<int> diff(maxLocation, 0);

    // 更新差分数组
    for (const auto&amp; trip : trips) {
        diff[trip[1]] += trip[0]; // 乘客在 fromi 上车
        diff[trip[2]] -= trip[0]; // 乘客在 toi 下车
    }

    // 模拟行车过程，并实时检查是否超过最大载客量
    int currentPassengers = 0;
    for (int i = 0; i < maxLocation; ++i) {
        currentPassengers += diff[i];
        if (currentPassengers > capacity) {
            return false; // 在某个时刻超过了最大载客量
        }
    }

    return true;
}

int main() {
    std::vector<std::vector<int>> trips = {{4, 5, 6}, {6, 4, 7}, {4, 3, 5}, {2, 3, 5}};
    int capacity = 13;
    bool result = carPooling(trips, capacity);
    std::cout << std::boolalpha << result << std::endl; // 输出 true

    return 0;
}