POJ-2406(KMP+字符串压缩)

Power String

POJ-2406

字符串压缩模板题，但是是求有多少个这样最短的子串可以组成s。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
using namespace std;
const int N=10000001;
int pi[N];
void Pi(string s){
    memset(pi,0,sizeof(pi));
    int n=s.length();
    pi[0]=0;
    for(int i=1;i<n;i++){
        int j=pi[i-1];
        while(j>0&&s[i]!=s[j]){
            j=pi[j-1];
        }
        if(s[i]==s[j])
            j++;
        pi[i]=j;
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    while(cin>>s){
        if(s==".")
            break;
        Pi(s);
        int n=s.length();
        int k=n-pi[n-1];
        if(n%k==0&&pi[n-1]>0){
            cout<<n/k<<endl;
        }else
        {
            cout<<1<<endl;
        }
    }
    return 0;
}