0514：自由之路（★★）

Ian

2016-05-28 约 738 字预计阅读 2 分钟次阅读

力扣第 514 题

题目

电子游戏“辐射4”中，任务 “通向自由” 要求玩家到达名为 “Freedom Trail Ring” 的金属表盘，并使用表盘拼写特定关键词才能开门。

给定一个字符串 ring ，表示刻在外环上的编码；给定另一个字符串 key ，表示需要拼写的关键词。您需要算出能够拼写关键词中所有字符的最少步数。

最初，ring 的第一个字符与 12:00 方向对齐。您需要顺时针或逆时针旋转 ring 以使 key 的一个字符在 12:00 方向对齐，然后按下中心按钮，以此逐个拼写完 key 中的所有字符。

旋转 ring 拼出 key 字符 key[i] 的阶段中：

您可以将 ring 顺时针或逆时针旋转 一个位置 ，计为1步。旋转的最终目的是将字符串 ring 的一个字符与 12:00 方向对齐，并且这个字符必须等于字符 key[i] 。
如果字符 key[i] 已经对齐到12:00方向，您需要按下中心按钮进行拼写，这也将算作 1 步。按完之后，您可以开始拼写 key 的下一个字符（下一阶段）, 直至完成所有拼写。

示例 1：

输入: ring = "godding", key = "gd"
输出: 4
解释:
对于 key 的第一个字符 'g'，已经在正确的位置, 我们只需要1步来拼写这个字符。
对于 key 的第二个字符 'd'，我们需要逆时针旋转 ring "godding" 2步使它变成 "ddinggo"。
当然, 我们还需要1步进行拼写。
因此最终的输出是 4。

示例 2:

输入: ring = "godding", key = "godding"
输出: 13

提示：

1 <= ring.length, key.length <= 100
ring 和 key 只包含小写英文字母
保证字符串 key 一定可以由字符串 ring 旋转拼出

分析

#1

按转到哪个位置递推即可，只需要顺/逆时针找第一个即可。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17


class Solution:
    def findRotateSteps(self, ring: str, key: str) -> int:
        n = len(ring)
        f = {0:0}
        for c in key:
            g = defaultdict(lambda:inf)
            for i,w in f.items():
                r = i
                while ring[r]!=c:
                    r = (r+1)%n
                g[r] = min(g[r],1+w+(r-i)%n)
                l = i
                while ring[l]!=c:
                    l = (l-1)%n
                g[l] = min(g[l],1+w+(i-l)%n)
            f = g
        return min(f.values())

114 ms

#2

可以预先保存下标列表，二分查找最近的下标。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17


class Solution:
    def findRotateSteps(self, ring: str, key: str) -> int:
        n = len(ring)
        d = defaultdict(list)
        for i,c in enumerate(ring):
            d[c].append(i)
        f = {0:0}
        for c in key:
            A = d[c]
            g = defaultdict(lambda:inf)
            for i,w in f.items():
                pos = bisect_left(A,i)
                l,r = A[pos-1],A[pos%len(A)]
                g[r] = min(g[r],1+w+(r-i)%n)
                g[l] = min(g[l],1+w+(i-l)%n)
            f = g
        return min(f.values())

72 ms

目录

0514：自由之路（★★）

题目

分析

#1

#2

解答