bzoj1090: [SCOI2003]字符串折叠 dp

可以先求每个区间可以折叠的长度，这个可以先hash一下

然后直接记忆化搜索。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
char st[110];
#define pb(a) push_back(a)
int x=123;
unsigned long long ha[110],p[110];
int n,dp[110][110];
vector<int>val[110][110];
int cal(int k)
{
    int tmp=0;
    while(k)
    {
        k/=10;
        tmp++;
    }
    return tmp;
}
bool isok(int i,int j,int len)
{
    unsigned long long a=ha[i]-ha[i+len]*p[len];
    unsigned long long b=ha[j]-ha[j+len]*p[len];
    return a==b;
}
bool judge(int l,int r,int len)
{
    int num=(r-l+1)/len;
    for(int j=1;j<num;j++)
    {
        if(!isok(l,l+j*len,len))
        {
            return false;
        }
    }
    return true;
}
void getval(int l,int r)
{
    int num=r-l+1;
    if(judge(l,r,1)) val[l][r].pb(num/1);
    for(int i=2;i*i<=num;i++)
    {
        if(num%i==0)
        {
            if(judge(l,r,i)) val[l][r].pb(num/i);
            if(judge(l,r,num/i)&&num!=i*i) val[l][r].pb(i);
        }
    }
}
int dfs(int l,int r)
{
    if(l==r) return 1;
    if(dp[l][r]) return dp[l][r];
    int num=(r-l+1);
    int tmp=num;
    for(int i=0;i<val[l][r].size();i++)
    {
        int k=val[l][r][i];
        tmp=min( tmp , dfs(l,l+(num/k)-1)+2+cal(k));
    }
    for(int i=l;i<r;i++)
    {
        tmp=min(tmp,dfs(l,i)+dfs(i+1,r));
    }
    dp[l][r]=tmp;
    return tmp;
}
int main()
{
    scanf("%s",st+1);
    n=strlen(st+1);
    ha[n+1]=0;
    for(int i=n;i;i--) ha[i]=ha[i+1]*x+st[i];
    p[0]=1;
    for(int i=1;i<=n;i++) p[i]=p[i-1]*x;
 
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            getval(i,j);
        }
    }
    printf("%d\n",dfs(1,n));
    return 0;
}