回文串+回溯法 URAL 1635 Mnemonics and Palindromes

题目传送门

 /*
     题意：给出一个长为n的仅由小写英文字母组成的字符串，求它的回文串划分的元素的最小个数，并按顺序输出此划分方案
     回文串+回溯：dp[i] 表示前i+1个字符(从0开始)最少需要划分的数量，最大值是i+1，即单个回文串；
         之前设置ok[j][j+i] 判断从j到j+i的字符是否为回文串(注意两个for的顺序，为满足ok[j][j+i] = ok[j+1][j+i-1])
         最后枚举找到最优划分点，用pre[i]记录前一个位置，print函数 输出空格
 */
 #include <cstdio>
 #include <iostream>
 #include <algorithm>
 #include <cmath>
 #include <cstring>
 using namespace std;

 const int MAXN = 4e3 + ;
 const int INF = 0x3f3f3f3f;
 int dp[MAXN];
 int pre[MAXN];
 bool ok[MAXN][MAXN];
 char s[MAXN];

 void print(int p)
 {
     if (pre[p] == -)
     {
         for (int i=; i<=p; ++i)    printf ("%c", s[i]);
     }
     else
     {
         print (pre[p]);    printf (" ");
         for (int i=pre[p]+; i<=p; ++i)    printf ("%c", s[i]);
     }
 }

 void work(void)
 {
     int len = strlen (s);

     for (int i=; i<len; ++i)
     {
         for (int j=; j+i<len; ++j)
         {
             if (i == )    ok[j][j+i] = true;
             else if (i == )    ok[j][i+j] = (s[j] == s[j+i]);
             else if (s[j] == s[j+i])    ok[j][j+i] = ok[j+][j+i-];
         }
     }

     for (int i=; i<len; ++i)
     {
         dp[i] = i + ;    pre[i] = i - ;
         if (ok[][i])    {dp[i] = ;    pre[i] = -;    continue;}
         for (int j=i-; j>=; --j)
         {
             if (ok[j+][i])
             {
                 if (dp[i] > dp[j] + )
                 {
                     dp[i] = dp[j] + ;    pre[i] = j;
                 }
             }
         }
     }

     printf ("%d\n", dp[len-]);
     print (len-);    puts ("");
 }

 int main(void)        //URAL 1635 Mnemonics and Palindromes
 {
     //freopen ("P.in", "r", stdin);

     while (scanf ("%s", s) == )
     {
         memset (ok, false, sizeof (ok));
         memset (dp, , sizeof (dp));
         memset (pre, , sizeof (pre));

         work ();
     }

     return ;
 }