LeetCode 每日一题 Day1

LancerWithX2023-12-05 23:35

1094. 拼车

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

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

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

示例 1：

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

输出：false

示例 2：

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

输出：true

提示：

1 <= trips.length <= 1000

trips[i].length == 3

1 <= numPassengers~i~ <= 100

0 <= fromi < to~i~ <= 1000

1 <= capacity <= 10^5^

我的代码实现如下，使用了差分法：

#include <iostream>
#include <vector>

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

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

    // 更新差分数组
    for (const auto& 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;
}

在这里给出了完整代码，至于使用差分法，是因为它可以高效处理数组元素的区间修改，正好与该题对应,使用迭代差分即可ac