解法
首先,把车按距终点的远近排。那么,一号车肯定是它属于的车队最后到终点时的状态。
- 如果二号车追不上它,那么一号车自己一个车队,问题转化成少了一号车的另一个小问题。
- 假如二号车能追上一号车,那么一号跟二号就一个车队了,这时候对于三号车来说:
要注意,如果三号车能追上二号车,那它一定能追上一号车,它的逆否命题是,如果三号车追不上一号车,那么它一定追不上二号车。
所以,对三号车来说,只要它能追上一号车,那么它就是一号车车队里的,否则就不是
这个命题同样可以拓展到一号车车队里不止二号车这一辆车的场景
可以看出,某个车是被合并还是单独在一个新的车队取决于它前一个车队的leader
class Solution(object):
def carFleet(self, target, position, speed):
"""
:type target: int
:type position: List[int]
:type speed: List[int]
:rtype: int
"""
n = len(position)
from collections import deque
stack = deque()
rank = sorted(xrange(n),key=lambda x:-position[x])
# print rank
for i in rank:
if not stack or speed[i]*(target-position[stack[-1]])<(target-position[i])*speed[stack[-1]]:
stack.append(i)
# print stack
return len(stack)