题目:
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
解答:
方法一:暴力法
代码一:
class Solution {
public boolean carPooling(int[][] trips, int capacity) {
int[] count=new int[1000];
for(int i=0;i<trips.length;i++){
for(int j=trips[i][1];j<trips[i][2];j++){
count[j]=count[j]+trips[i][0];
}
}
int max=0;
for(int i=0;i<count.length;i++){
if(max<count[i]){
max=count[i];
}
}
return max<=capacity;
}
}
方法二:
代码二:
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;
}
}