【Leetcode】818. Race Car 818. 赛车
解法
令表示连续的k个A,容易分析出,它会在该方向上走步,我们可以将最后的结果表示为,最后一个R可以去掉
它用数学表达式写出来就是:
有个结论:
设K为target的二进制位数,那么有
如果要超过target后再往回走,最多会走到的位置,不可以再往后走了,因为假如再多走S,正向一步走了S,反向要抵消这一步需要至少步。
解法一:DIJSTRA
我们把图里的每个点i表示为【沿该方向向target走还差i距离】
初始为正向,那么,所以起点是dist[target]=0
。
从点i可以去哪些点呢?我们由上文的分析可以知道,所以从点i可以通过走、、、…步,然后再掉头,即是附加、、…,它们分别增加1、2、3、…、K+1步。
在没有加R的时候,剩余的距离为,由于掉头了,那么剩余的距离变成,因为要倒着走才能到target了。
所以每个点i将会联通K+1个点。
现在我们用dijstra算法,每次取步数最少的那个点,它就是真正到点i的最少步数,然后再看看从点i走的点是否能改善距离。
需要注意的是,数轴上每个整数位置都可能成为点i,这样循环就不会停了,这时候要用上上面的结论,我们的车不可能跑出,这就让点的个数变得有限了。
class Solution(object):
def racecar(self, target):
"""
:type target: int
:rtype: int
"""
K = target.bit_length()
up = (1<<K)-1
from heapq import heappush,heappop
queue = [(0,target)]
dist = {target:0}
while queue:
# print queue
d,t = heappop(queue)
if t in dist and dist[t]<d:continue
for k in xrange(K+1):
d2,t2 = d+k+1,(1<<k)-1-t
if t2==0:
d2-=1
if abs(t2)<=up and (t2 not in dist or dist[t2]>d2):
dist[t2]=d2
heappush(queue,(d2,t2))
return dist[0]
解法二:DP
令表示走t距离用的最小步数
我们走到一个target,有三种情况:
-
如果target刚好等于,那么就刚好要走K步。
-
永远不超过这个target,走到一个位置,反回去几步,再向前走。我们一定要正向走到不能再走,即走到处。因为:
首先,从上面的表达式我们知道,每个小节的顺序不影响结果。那么我们可以假设,下标同为偶数的是递减排列的。
那么,此时最大的一定是K-1,否则,假如是更小的值k,从到这段距离用后面更更小的来走,肯定花费的步数更多。走到处再掉头,花费K步()。然后我们需要枚举往回走j步()再掉头,花费j+1步()。现在我们离target的距离缩短为,即,它要花费步,所以总花费步数为
-
超过这个target,然后往回走。从上面的结论可知,我们最多走到的位置,即花费K+1步()。
然后我们就要往回走距离,这个距离要花步。
总共花费的步数是
综上所述,那么就在上面的所有候选者里选最小值
class Solution(object):
def racecar(self, target):
"""
:type target: int
:rtype: int
"""
INF = 3*target
f = [INF]*(target+1)
f[0] = 0
for t in xrange(1,target+1):
k = t.bit_length()
if t+1==1<<k:
f[t]=k
continue
u = t-(1<<(k-1))
for j in xrange(k-1):
f[t]=min(f[t],f[u+(1<<j)]+k+j+1)
if (1<<k)-t<t:
f[t] = min(f[t],k+1+f[(1<<k)-t])
return f[target]