Educational Codeforces Round 25 F. String Compression(kmp+dp)

 1 #include<bits/stdc++.h>
 2 #define F(i,a,b) for(int i=a;i<=b;++i)
 3 using namespace std;
 4 
 5 void kmp_pre(char *x,int m,int *nxt){
 6     int i,j;
 7     j=nxt[0]=-1,i=0;
 8     while(i<m){
 9         while(-1!=j&&x[i]!=x[j])j=nxt[j];
10         nxt[++i]=++j;
11     }
12 }
13 const int N=10000;
14 int len,nxt[N],dp[N],cnt[N];
15 char s[N],buf[20],str[N];
16 
17 int main(){
18     scanf("%s",s);
19     len=strlen(s);
20     F(i,1,len)cnt[i]=sprintf(buf,"%d",i),dp[i]=i+1;
21     F(i,0,len-1)
22     {
23         strcpy(str,s+i);
24         kmp_pre(str,len-i,nxt);
25         for(int j=1;i+j<=len;j++)//枚举长度
26         {
27             int L=j-nxt[j];//以j结尾的子串的最小循环节
28             if(j%L==0)//如果这个子串的长度恰好能整除最小循环节，那么可以按照最小循环节压缩。
29             {
30                 dp[i+j]=min(dp[i+j],dp[i]+L+cnt[j/L]);
31             }else dp[i+j]=min(dp[i+j],dp[i]+j+cnt[1]);
32         }
33     }
34     printf("%d\n",dp[len]);
35     return 0;
36 }