-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path853.py
36 lines (34 loc) · 1.35 KB
/
853.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution:
def carFleet(self, target: int, position: List[int], speed: List[int]) -> int:
cars = []
for i in range(len(position)):
cars.append((position[i], speed[i]))
cars = sorted(cars, reverse=True)
result = []
for car in cars:
if len(result) == 0:
result.append(car)
continue
leadPos, leadSpeed = result.pop()
leadSteps = (target-leadPos)/leadSpeed
carSteps = (target-car[0])/car[1]
if carSteps <= leadSteps:
result.append((leadPos, leadSpeed))
else:
result.append((leadPos, leadSpeed))
result.append(car)
return len(result)
# improvemets:
# * compact cars population
# * in result store seconds
# * compare last two and if faster pop()
class Solution:
def carFleet(self, target: int, position: List[int], speed: List[int]) -> int:
cars = [(p, s) for p, s in zip(position, speed)]
cars = sorted(cars, reverse=True)
result = []
for car in cars:
result.append((target-car[0])/car[1])
if len(result) >= 2 and result[-1] <= result[-2]:
result.pop()
return len(result)