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 <= numPassengersi <= 100
0 <= fromi < toi <= 1000
1 <= capacity <= 105
我的代码实现如下,使用了差分法:
#include <iostream>
#include <vector>bool carPooling(std::vector<std::vector<int>>& trips, int capacity) {const int maxLocation = 1001; // 最大位置数,根据题目要求设定// 创建差分数组,初始化为0std::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; // 输出 truereturn 0;
}
在这里给出了完整代码,至于使用差分法,是因为它可以高效处理数组元素的区间修改,正好与该题对应,使用迭代差分即可ac