题目描述:
假设你是一位顺风车司机,车上最初有 capacity 个空座位可以用来载客。由于道路的限制,车 只能 向一个方向行驶(也就是说,不允许掉头或改变方向,你可以将其想象为一个向量)。
这儿有一份行程计划表 trips[][],其中 trips[i] = [num_passengers, start_location, end_location] 包含了你的第 i 次行程信息:
必须接送的乘客数量;
乘客的上车地点;
以及乘客的下车地点。
这些给出的地点位置是从你的 初始 出发位置向前行驶到这些地点所需的距离(它们一定在你的行驶方向上)。
请你根据给出的行程计划表和车子的座位数,来判断你的车是否可以顺利完成接送所用乘客的任务(当且仅当你可以在所有给定的行程中接送所有乘客时,返回 true,否则请返回 false)。
示例 1:
输入:trips = [[2,1,5],[3,3,7]], capacity = 4
输出:false
示例 2:
输入:trips = [[2,1,5],[3,3,7]], capacity = 5
输出:true
示例 3:
输入:trips = [[2,1,5],[3,5,7]], capacity = 3
输出:true
示例 4:
输入:trips = [[3,2,7],[3,7,9],[8,3,9]], capacity = 11
输出:true
提示:
你可以假设乘客会自觉遵守 “先下后上” 的良好素质
trips.length <= 1000
trips[i].length == 3
1 <= trips[i][0] <= 100
0 <= trips[i][1] < trips[i][2] <= 1000
1 <= capacity <= 100000
我的代码:
按照正常的思路来进行,每次经过一个车站,都计算要过往下车的人数和当前上车的人数,判断是否大于容量capacity。
public static boolean carPooling(int[][] trips, int capacity) {
List<int[]> list = new ArrayList<>();
for (int[] is : trips) {
list.add(is);
}
Collections.sort(list,new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if(o1[1] == o2[1]){
return -(o1[2] - o2[2]);
}
else {
return o1[1] - o2[1];
}
}
});
// 表示的是哪一站要下车多少人
Map<Integer, Integer> map = new TreeMap();
int end = list.get(list.size() - 1)[2];
int start = 1;
int index = 0;
// 下车
int pa = 0;
int temstart = 1;
while (index < list.size()) {
int tem[] = list.get(index);
System.out.println(Arrays.toString(tem));
start = tem[1];
//该站有下车的乘客
while (temstart <= start) {
capacity += map.getOrDefault(temstart, 0);
temstart ++;
}
pa = 0;
// 查看上车的人数
if(capacity < tem[0]){
return false;
}
capacity -= tem [0];
map.put(tem[2], map.getOrDefault(tem[2], 0) + tem [0]);
index ++;
}
return true;
}
还有以一种思路:
就是先计算每个车站的上车人数:包括下车和上车的(可能为负数),之后我们依次进行遍历,进行加和,判断是否大于capacity。
这就不需要先排序了
代码:
public static boolean carPooling(int[][] trips, int capacity) {
Map<Integer, Integer> map = new TreeMap<>();
for (int[] is : trips) {
map.put(is[1], map.getOrDefault(is[1], 0) + is[0]);
map.put(is[2], map.getOrDefault(is[2], 0) - is[0]);
}
int tem = 0;
for (Entry<Integer, Integer> is : map.entrySet()) {
tem += is.getValue();
if(tem > capacity){
return false;
}
}
return true;
}
是不是很简单了…