【教3妹学编程-算法题】拼车
3妹:“太阳当空照,花儿对我笑,小鸟说早早早,你为什么背上炸药包”
2哥 :3妹,什么事呀这么开发。
3妹:2哥你看今天的天气多好啊,阳光明媚、万里无云、秋高气爽,适合秋游。
2哥:是啊,立冬之后天气多以多云为主,好不容易艳阳高照。可是你不能秋游,赶紧收拾收拾上班去啦
3妹:哼, 好吧~
2哥:给你出了一道题发你微信里了, 上班通勤的路上记得看一下,回来问你答案~
3妹:知道啦,难不倒我!
题目:
车上最初有 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 <= 10^5
思路:
思考差分,
从朴素的想法开始:创建一个数组 cnt,用于存储从某个站点出发时,车上的乘客数量。
例如 cnt[x]=c含义为在站点 x 出发时(在该站点的下车和上车均完成),车上乘客数为 c 个。
对于每个 trips[i]=(c,a,b),我们需要对 [a,b)范围内的 cnt[j]进行加 c 操作。
java代码:
class Solution {
public boolean carPooling(int[][] trips, int capacity) {
int toMax = 0;
for (int[] trip : trips) {
toMax = Math.max(toMax, trip[2]);
}
int[] diff = new int[toMax + 1];
for (int[] trip : trips) {
diff[trip[1]] += trip[0];
diff[trip[2]] -= trip[0];
}
int count = 0;
for (int i = 0; i <= toMax; ++i) {
count += diff[i];
if (count > capacity) {
return false;
}
}
return true;
}
}