力扣原文力扣题解
514. 自由之路
难度困难221
电子游戏“辐射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 的下一个字符(下一阶段), 直至完成所有拼写。
示例:
输入: ring = "godding", key = "gd"
输出: 4
解释:
对于 key 的第一个字符 'g',已经在正确的位置, 我们只需要1步来拼写这个字符。
对于 key 的第二个字符 'd',我们需要逆时针旋转 ring "godding" 2步使它变成 "ddinggo"。
当然, 我们还需要1步进行拼写。
因此最终的输出是 4。
提示:
- ring 和 key 的字符串长度取值范围均为 1 至 100;
- 两个字符串中都只有小写字符,并且均可能存在重复字符;
- 字符串 key 一定可以由字符串 ring 旋转拼出。
解析:
**原题可以转化为:圆盘固定,我们可以拨动*指针;*现在需要我们拨动指针并按下按钮,以最少的操作次数输入key
对应的字符串。
- 「状态」就是「当前需要输入的字符」和「当前圆盘指针的位置」。
再具体点,「状态」就是i
和j
两个变量。我们可以用i
表示当前圆盘上指针指向的字符(也就是ring[i]
);用j
表示需要输入的字符(也就是key[j]
)。
这样我们可以写这样一个dp
函数:
int dp(string& ring, int i, string& key, int j);
-
这个
dp
函数的定义如下:当圆盘指针指向
ring[i]
时,输入字符串key[j..]
至少需要dp(ring, i, key, j)
次操作。计算dp(ring, 0, key, 0)
的值 -
basecase
int dp(string& ring, int i, string& key, int j) { // base case,完成输入 if (j == key.size()) return 0; // ... }
-
「选择」就是「如何拨动指针得到待输入的字符」。
比如说输入
ring = "gdonidg"
,现在圆盘的状态如下图:
假设我想输入的字符key[j] = "d"
,圆盘中有两个字母"d"
,而且我可以顺时针也可以逆时针拨动指针,所以总共有四种「选择」输入字符"d"
,我们需要选择操作次数最少的那个拨法。
但是对于圆盘中的两个字符"d"
,还能是怎么近怎么来吗?
不能,因为这和key[i]
之后需要输入的字符有关
如果输入的是key = "di"
,那么即便右边的"d"
离得近,也应该去左边的"d"
,因为左边的"d"
旁边就是"i"
,「整体」的操作数最少。
那么,应该如何判断呢?其实就是穷举,递归调用dp
函数,把两种选择的「整体」代价算出来,然后再做比较就行了
其中用map保存同一值的不同位置,然后做选择时遍历这些位置,与其他元素加一起的整体代价算出来,选择最小的代价
代码:
class Solution {
// 字符 -> 索引列表
HashMap<Character,List<Integer>> charToIndex=new HashMap<Character,List<Integer>>();
// 备忘录
int[][] memo;
public int findRotateSteps(String ring, String key) {
int m=ring.length();
int n=key.length();
memo=new int[m+1][n+1];
//备忘录全部初始化为0
for(int[] row:memo){
Arrays.fill(row,0);
}
//记录圆环上字符到索引的映射
for(int i=0;i<ring.length();i++){
char c=ring.charAt(i);
List<Integer> list=charToIndex.get(c);
if (list == null) {
list = new ArrayList<>();
charToIndex.put(c, list);
}
list.add(i);
}
return dp(ring,0,key,0);
}
// 计算圆盘指针在 ring[i],输入 key[j..] 的最少操作数
public int dp(String ring,int i,String key,int j){
int n=ring.length();
//basecase,完成输入
if(j==key.length()) return 0;
// 查找备忘录,避免重叠子问题
if(memo[i][j]!=0) return memo[i][j];
// 做选择
int res=Integer.MAX_VALUE;
// ring 上可能有多个字符 key[j]
for(int k : charToIndex.get(key.charAt(j))){
//拨动指针的次数
int delta=Math.abs(i-k);
//选择顺时针还是逆时针
delta=Math.min(delta,n-delta);
// 将指针拨到 ring[k],继续输入 key[j+1..]
int subproblem=dp(ring,k,key,j+1);
res=Math.min(res,1+delta+subproblem);
}
// 将结果存入备忘录
memo[i][j]=res;
return res;
}
}