51NOD 1154 回文串划分 【DP+Manacher】

1154 回文串划分
基准时间限制：1 秒 空间限制：131072 KB 分值: 40 难度：4级算法题 收藏  关注
有一个字符串S，求S最少可以被划分为多少个回文串。
例如：abbaabaa，有多种划分方式。

a|bb|aabaa - 3 个回文串
a|bb|a|aba|a - 5 个回文串
a|b|b|a|a|b|a|a - 8 个回文串

其中第1种划分方式的划分数量最少。
Input
输入字符串S(S的长度<= 5000)。
Output
输出最少的划分数量。
Input示例
abbaabaa
Output示例
3

---

考虑Dp一下

d[i]=str[0-i]最少能分成多少回文串
d[i]=d[i-1]+1
d[i]=min(d[j]+1) | str[j,i]为回文串

O(n^3)复杂度 不理想
如果能在快速确定str[j,i]是否是回文 就OK了

跑一遍Manacher
RI[i]-1是以i位置为中心的回文串的最大长度(填充了#的回文串)

bool isP(int s,int e){//返回 原str中 下标为s到e的子串是否为回文
    int len=e-s+1;
    s=2*s+1,e=2*e+1;
    int center=(s+e)/2;
    return RI[center]-1>=len;
}

每次查询str[j,i]为O(1)
复杂度下降为O(n^2)

---

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
#include<queue>
#include<sstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>

const int N = 5000+5;

char str[N];
char tStr[2*N];
int RI[2*N];

int manacher(char*str){//求最长回文子串
    int n=strlen(str);
    int len=0;
    tStr[len++]='#';
    for(int i=0;i<n;++i){
        tStr[len++]=str[i];
        tStr[len++]='#';
    }
    RI[0]=1;
    int maxRi=0,pos=0;//扩张到的最右边的下标，对称中心下标
    int ans=0;
    for(int i=1;i<len;++i){
        if(i<maxRi){
            RI[i]=min(RI[2*pos-i],maxRi-i);
        }
        else{
            RI[i]=1;
        }

        for(int l=i-RI[i],r=i+RI[i];l>=0&&r<len;--l,++r){//向2边扩张
            if(tStr[l]!=tStr[r]){
                break;
            }
            ++RI[i];
        }

        if(i+RI[i]-1>maxRi){
            maxRi=i+RI[i]-1;
            pos=i;
        }
        ans=max(RI[i]-1,ans);
    }
    return ans;
}

bool isP(int s,int e){//返回 原str中 下标为s到e的子串是否为回文
    int len=e-s+1;
    s=2*s+1,e=2*e+1;
    int center=(s+e)/2;
    return RI[center]-1>=len;
}

int d[N];//d[i]=以str[0-i]能分成最少多少个回文

int slove(){
    manacher(str);
    int n=strlen(str);
    fill(d,d+n+1,INF);
    d[0]=1;
    for(int i=1;i<n;++i){
        d[i]=d[i-1]+1;
        for(int j=0;j<i;++j){
            if(isP(j,i)){
                d[i]=min(1+(j==0?0:d[j-1]),d[i]);
            }
        }
    }
    return d[n-1];
}

int main()
{
    //freopen("/home/lu/Documents/r.txt","r",stdin);
    //freopen("/home/lu/Documents/w.txt","w",stdout);
    scanf("%s",str);
    printf("%d\n",slove());
    return 0;
}