LeetCode 214. Shortest Palindrome

#include 
#include 
#include 
#include 
using namespace std;

/*
  Given a string S, you are allowed to convert it to a palindrome by adding
  characters in from of it. Find and return the shortest palindrome you
  can find by performing this transformation.

  For example:
  Given "aacecaaa", return "aaacecaaa"
  Given "abcd", return "dcbabcd"
*/

string shortestPalindrome(string s) {
  string t = s;
  reverse(t.begin(), t.end());
  string tmp = t + '#' + s;
  int n = tmp.size();
  vector dp(n, 0);
  for(int i = 1; i <= tmp.size(); ++i) {   // KMP algorithm
    int j = dp[i-1];
    while(j > 0 && (tmp[i] != tmp[j])) {
      j = dp[j-1];
    }
    dp[i] = (j + (tmp[i] == tmp[j]));
  }
  return t.substr(0, s.size() - dp[tmp.size() - 1]) + s;
}

int main(void) {
  cout << shortestPalindrome("abc") << endl;
}

你可能感兴趣的:(LeetCode,题解,Multiple,Passes)