题目地址:
https://leetcode.com/problems/car-pooling/
在一个数轴上,从 0 0 0开始,一辆容量为 c c c的车向右行驶,给定若干乘客的上车点和下车点,以若干三元组表示,第一维是乘客数量,后面两维是上车坐标和下车坐标,车不回头,只能向右走。问这辆车能否将每位乘客从上车点运到下车点。题目保证上下车的点的范围是 [ 0 , 1000 ] [0,1000] [0,1000]。
思路是差分数组。这题实际上要实现某个区间加上某个数 x x x的操作,可以用差分数组在 O ( 1 ) O(1) O(1)时间实现,最后只需要还原成原数组然后看一下是否每个点都不超过 c c c即可。代码如下:
class Solution {
public:
bool carPooling(vector<vector<int>>& trips, int capacity) {
int diff[1001] = {0};
for (auto& trip : trips) diff[trip[1]] += trip[0], diff[trip[2]] -= trip[0];
int sum = 0;
for (int x : diff) {
sum += x;
if (sum > capacity) return false;
}
return true;
}
};
时间复杂度 O ( n ) O(n) O(n)( n n n为三元组的数量),空间 O ( M ) O(M) O(M)( M M M为上下车点的最大范围)。