一、题目
N 辆车沿着一条车道驶向位于 target 英里之外的共同目的地。
每辆车 i 以恒定的速度 speed[i] (英里/小时),从初始位置 position[i] (英里) 沿车道驶向目的地。
一辆车永远不会超过前面的另一辆车,但它可以追上去,并与前车以相同的速度紧接着行驶。
此时,我们会忽略这两辆车之间的距离,也就是说,它们被假定处于相同的位置。
车队 是一些由行驶在相同位置、具有相同速度的车组成的非空集合。注意,一辆车也可以是一个车队。
即便一辆车在目的地才赶上了一个车队,它们仍然会被视作是同一个车队。
会有多少车队到达目的地?
示例:
输入:target = 12, position = [10,8,0,5,3], speed = [2,4,1,1,3]
输出:3
解释:
从 10 和 8 开始的车会组成一个车队,它们在 12 处相遇。
从 0 处开始的车无法追上其它车,所以它自己就是一个车队。
从 5 和 3 开始的车会组成一个车队,它们在 6 处相遇。
请注意,在到达目的地之前没有其它车会遇到这些车队,所以答案是 3。
提示:
0 <= N <= 10 ^ 4
0 < target <= 10 ^ 6
0 < speed[i] <= 10 ^ 6
0 <= position[i] < target
所有车的初始位置各不相同。
链接:https://leetcode-cn.com/problems/car-fleet
二、思路
提示中10的4次方,时间复杂度有要求,需要以空间换时间。注意double类型除法,需要除数为double类型
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.TreeMap;
public class Carfleet {
public static int carFleet(int target, int[] position, int[] speed) {
//10 [0,4,2] [2,1,3]
int len = position.length;
double[] minute =new double[len];
for (int i = 0; i < len; i++){
minute[i] = (target - position[i]) / (double)speed[i];
}
//额外定义空间换时间,避免超时
Map<Integer,Double> map =new TreeMap<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return Integer.compare(o1,o2);
}
});
for (int i = 0; i < len; i++){
map.put(position[i],minute[i]);
}
int index = 0;
for (int pos : map.keySet()){
position[index] = pos;
minute[index] = map.get(pos);
index++;
}
int count = 0;
//额外定义空间换时间
boolean[] visited = new boolean[len];
for (int i = len - 1; i >= 0; i--){
if (visited[i]){
continue;
}
visited[i] = true;
count++;
for (int j = i - 1; j >= 0; j--){
if (minute[j] <= minute[i]){
visited[j] = true;
}
}
}
return count;
}
public static void main(String[] args) {
System.out.println(carFleet(10,new int[]{0,4,2},new int[]{2,1,3})) ;
}
}