给定一个字符串 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 的下一个字符(下一阶段), 直至完成所有拼写。
来源,leetcode 每日一题 514. 自由之路
例如:
输入: ring = "godding", key = "gd"输出: 4解释:对于 key 的第一个字符 'g',已经在正确的位置, 我们只需要1步来拼写这个字符。对于 key 的第二个字符 'd',我们需要逆时针旋转 ring "godding" 2步使它变成 "ddinggo"。当然, 我们还需要1步进行拼写。因此最终的输出是 4。
解题思路
- 很容易想到的解决思路是:记录一个当前位置,然后遍历每一个可以key中的下一个字符相匹配的地方,当key中字符都匹配完成之后,选取一个最小的数值作为答案。此方法的复杂度为
,复杂度太高,无法通过。
- 需要对上面的过程进行优化
- 将所有单词的出现的位置都记录下来,然后每次可以根据下一个字符是什么直接找到它的位置
- 记忆化:计算由上一个字符到达当前字符所在位置的最小值,然后保存下来
代码
class Solution {public:int findRotateSteps(string ring, string key) {map<char, vector<int>> char_indexs;vector<int> last_pos(1, 0);vector<int> last_val(1, 0);for (int i = 0; i < ring.length(); i++) {char_indexs[ring[i]].push_back(i);}int result = INT_MAX;int length = ring.length();int length2 = key.length();dfs(result, char_indexs, key, 0, last_pos, length, length2, last_val);return result + key.length();}void dfs(int& result, map<char, vector<int>>& char_indexs, string& key, int curr_char, vector<int> last_pos, int& length, int& length2, vector<int> last_value){if (curr_char == length2) {for (auto& item: last_value) {result = min(result, item);}return;}vector<int> temp;vector<int> temp_val;for (auto& item:char_indexs[key[curr_char]]) {// 遍历当前字符的所在位置// 记录上一个字符到达当前字符的最短路径int mins = INT_MAX;// 对上一个字符的位置进行遍历for (int i = 0; i < last_pos.size(); i++) {int c_pos = last_pos[i];int value = last_value[i];int t = abs(item-c_pos);t = min(length-t, t);mins = min(t + value, mins);}temp_val.push_back(mins);temp.push_back(item);}dfs(result, char_indexs, key, curr_char + 1, temp, length, length2, temp_val);}};
