LeetCode 514: Freedom Trail 解题与思考

最新推荐文章于 2020-11-11 00:43:20 发布

cjl707408282

最新推荐文章于 2020-11-11 00:43:20 发布

阅读量446

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/cjl707408282/article/details/78397228

版权

LeetCode 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

LeetCode 514: Freedom Trail 解题与思考

[原题链接]

题目描述

我们有一个轮盘，轮盘上有一个首尾相接的字符串S，我们有如下操作
1、转轮盘到下一个或者上一个字符，需要一步操作
2、输出当前字符，需要一步操作
假如我们有目标字符串K，求取出K的最小操作步数

思路

第一感觉就是用DP，因为看上去除去末尾的“上一个字符串”所需要的距离，加上从上一个最小距离，然后求极小值，就是所需答案。
但是遇到了几个问题：

假设结束的字符为E，在轮盘中有若干个相同的E，你不知道以哪个结尾是最优解
你也不知道上一个字符串的末尾停留在哪个位置时能取到最优解

所以我们至少要把问题稍微做一下转换：

Step1
假设字符串 $K_n$ 长度为n，以 $E_n$ 结尾，有p个E( $E_{n, 1}, E_{n,2}, ...., E_{n,p}$ )(此处每个变量分别记录其所在位置)，那么，我们就是在寻找以某个位于 $E_{n,i}$ 结尾所得到的字符串K所用步数最少。
记字符串 $E_i$ 结尾所需要的步数为 $S_n(E_{n,i})$ ，记得到长度为n的字符串K的最小步数为 $K_n$ 即

得 到 K 的 最 少 步 数 = m i n (S n (E n, i)) - - ①

$得到K的最少步数 = min(S_n(E_{n,i})) --①$

Step2
然后这个时候我们需要求解 $S_n(E_{n,i})$ ，问题就变成：求解以某个特定位置 $E_{n,i}$ 结尾的得到K的最小步数。这个问题就是很明显的DP了。
不妨设上一个字符串为 $K_{n-1}$ ，结尾字符为 $E_{n-1}$ ，假设我们已经求得 $S_{n - 1}(E_{n - 1,j})$ ，也即以位置 $E_{n - 1,j}$ 结尾的时候所需的最小步数，那么我们只要找出从所有 $E_{n-1}$ 中出发到达 $E_{n,i}$ 中所用步数最少的，就是以特定位置 $E_{n,i}$ 结尾的最小步数，即

S n (E n, i) = m i n (S n - 1 (E n - 1, j) + E n - 1, j 到 E n, i 的 最 小 步 数) - - ②

$S_n(E_{n,i}) = min(S_{n - 1}(E_{n - 1,j}) + E_{n - 1,j}到E_{n,i}的最小步数) --②$

于是递推式就出来了。其中 $S_k$ 则为最优子问题。

算法

设计一个栈，栈内元素为二元对，记录了某个字符c的位置以及到达c耗费的最小步数

假设有两个这样的栈，一个为now，一个为last，分别记录当前结尾字符所在位置以及最小步数，和上一个结尾字符所在位置以及最小步数

初始化：我们将轮盘开始字符压now栈，最小步数为0；
以n记录取到字符串的第几个

循环做以下步骤：
1、将now的元素移动到last
2、对于字符串 $K_n$ ，假设结尾字符为 $E_n$ ，对每个 $E_{n,i}$ ，从last栈中取出所有的元素，按照②式与之计算最短路径，求最小值，然后压入now
3、n自加1

最后的解为now栈中所有最小路径中的最小值

代码

#include <iostream>
#include <vector>
#include <string>
using namespace std;

class Solution {
    int distance(int a, int b, int size) {
        return (a < b) ? ((b - a > size / 2) ? (size + a - b) : (b - a)) : ((a - b > size / 2) ? (size + b - a) : (a - b));
    }
public:
    int findRotateSteps(string ring, string key) {
        vector<vector<int>> charArr;
        vector<pair<int, int>> sol, solNext;
        sol.push_back(pair<int, int>(0, 0));
        int nowChar = 0;
        charArr.resize(26);
        int i = 0;
        for ( auto c : ring ) {
            charArr.at(c - 'a').push_back(i++);
        }
        while ( nowChar != key.size() ) {
            for ( auto pos : charArr.at(key.at(nowChar) - 'a') ) {
                pair<int, int> best(pos, 100000);
                for ( auto nowpos : sol ) {
                    int dis = distance(nowpos.first, pos, ring.size());
                    if ( nowpos.second + dis + 1 < best.second ) {
                        best.second = nowpos.second + dis + 1;
                    }
                }
                solNext.push_back(best);
            }
            sol = solNext;
            solNext.resize(0);
            nowChar++;
        }
        int answer = 100000;
        for ( auto ans : sol ) {
            answer = (ans.second < answer) ? ans.second : answer;
        }
        return answer;
    }
};