给定一个字符串 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);
}
};