假设你是一位顺风车司机,车上最初有 capacity 个空座位可以用来载客。由于道路的限制,车 只能 向一个方向行驶(也就是说,不允许掉头或改变方向,你可以将其想象为一个向量)。
这儿有一份乘客行程计划表 trips[][],其中 trips[i] = [num_passengers, start_location, end_location] 包含了第 i 组乘客的行程信息:
必须接送的乘客数量;
乘客的上车地点;
以及乘客的下车地点。
这些给出的地点位置是从你的 初始 出发位置向前行驶到这些地点所需的距离(它们一定在你的行驶方向上)。
请你根据给出的行程计划表和车子的座位数,来判断你的车是否可以顺利完成接送所有乘客的任务(当且仅当你可以在所有给定的行程中接送所有乘客时,返回 true,否则请返回 false)。
示例 1:
输入:trips = [[2,1,5],[3,3,7]], capacity = 4
输出:false
代码
class Solution {
public boolean carPooling(int[][] trips, int capacity) {
Map<Integer,Integer> add=new HashMap<>();
Map<Integer,Integer> sub=new HashMap<>();
TreeSet<Integer> treeSet=new TreeSet<>();
for(int[] c:trips)
{
treeSet.add(c[1]);//用treeset记录上下车的时间节点
treeSet.add(c[2]);
add.put(c[1],add.getOrDefault(c[1],0)+c[0]);//记录上车的时间点和人数
sub.put(c[2],sub.getOrDefault(c[2],0)+c[0]);
}
for(int c:treeSet)//模拟上下车
{
if(sub.containsKey(c)) capacity+=sub.get(c);
if(add.containsKey(c)) capacity-=add.get(c);
if(capacity<0) return false;
}
return true;
}
}
代码
class Solution {
public boolean carPooling(int[][] trips, int capacity) {
int[] track=new int[1001];//记录每一个节点的改变量
for(int[] c:trips)
{
track[c[1]]-=c[0];
track[c[2]]+=c[0];
}
for(int c:track)//遍历所有节点,如果容量出现不满足的情况,则返回false
{
capacity+=c;
if(capacity<0) return false;
}
return true;
}
}