Freedom Trail [LeetCode 514]

Description


In the video game Fallout 4, the quest “Road to Freedom” requires players to reach a metal dial called the “Freedom Trail Ring”, and use the dial to spell a specific keyword in order to open the door.

Given a string ring, which represents the code engraved on the outer ring and another string key, which represents the keyword needs to be spelled. You need to find the minimum number of steps in order to spell all the characters in the keyword.

Initially, the first character of the ring is aligned at 12:00 direction. You need to spell all the characters in the string key one by one by rotating the ring clockwise or anticlockwise to make each character of the string key aligned at 12:00 direction and then by pressing the centre button.

At the stage of rotating the ring to spell the key character key[i]:

  1. You can rotate the ring clockwise or anticlockwise one place, which counts as 1 step. The final purpose of the rotation is to align one of the string ring’s characters at the 12:00 direction, where this character must equal to the character key[i].

  2. If the character key[i] has been aligned at the 12:00 direction, you need to press the centre button to spell, which also counts as 1 step. After the pressing, you could begin to spell the next character in the key (next stage), otherwise, you’ve finished all the spelling.

Example:

Freedom Trail [LeetCode 514]_第1张图片

Input:
ring = “godding”, key = “gd”

Output:
4

Explanation:
For the first key character ‘g’, since it is already in place, we just need 1 step to spell this character.
For the second key character ‘d’, we need to rotate the ring “godding” anticlockwise by two steps to make it become “ddinggo”.
Also, we need 1 more step for spelling.
So the final output is 4.

Analysis


int size = ring.size();
int ksize = key.size();
vector<vector<int>> mp(26);
for(int i = 0; i < size; ++i){
    mp[ring[i]-'a'].push_back(i);
}

Freedom Trail [LeetCode 514]_第2张图片

vector<int> dp (size,INT_MAX);
dp[0] = 0;
vector<int> startIndex(1,0);  // starting index
for(int i=1;i<=ksize;++i){
    vector<int> nextDp(size,INT_MAX);
    for(auto it:mp[key[i-1]-'a']){
        for(int j=0;jint minDist = min((startIndex[j] + size -it)%size,(it + size - startIndex[j])%size) + dp[startIndex[j]];
            nextDp[it] =min(nextDp[it],minDist);
        }
    }
    startIndex = mp[key[i-1]-'a'];
    dp = nextDp;
} 
  • Start from the starting index 0, I compute all the distance between the starting index and the key[0]. Thus, we have dp as following.
    Freedom Trail [LeetCode 514]_第3张图片
  • Then I compute the distance between 0 (6) and key[1] separately, and add the distance from step one. We have the final answer as following.
    Freedom Trail [LeetCode 514]_第4张图片

Answer


#include 
#include 
#include 
#define INT_MAX 9999

using namespace std;
class Solution {
public:
    int findRotateSteps(string ring, string key) {
        int size = ring.length();
        int ksize = key.length();
        cout << size << " " << ksize << endl;
        vector<vector<int>> mp(26);
        for (int i = 0; i < size; i++) {
            mp[ring[i]-'a'].push_back(i);
        }

        vector<int> dp(size, INT_MAX);
        dp[0] = 0;
        vector<int> startindex(1, 0);
        for (int i = 0; i < ksize; i++) {
            vector<int> nextdp(size, INT_MAX);
            for (auto it:mp[key[i]-'a']) {
                for (int j = 0; j < startindex.size(); j++) {
                    int mindis = min((startindex[j]-it+size)%size, (it-startindex[j]+size)%size) + dp[startindex[j]];
                    nextdp[it] = min(nextdp[it], mindis);
                }
            }
            startindex = mp[key[i]-'a'];
            dp = nextdp;
        }
        int res = INT_MAX;
        for (int i = 0; i < size; i++) {
            res = min(res, dp[i]);
        }
        return res+ksize;
    }
};

你可能感兴趣的:(算法设计与分析)