1. 程式人生 > >poj2406 Power Strings kmp

poj2406 Power Strings kmp

      一個字串可以由他的一個子串a重複n次得到,現在給一個長度不大於10^6的字串,求最長n。

      跟白書上的period (訓練之南P213,例13)基本一樣甚至還簡單一點。核心就是利用KMP中的失配函式求最短迴圈節。可以得到對於字串的第i位,若有i % (i-f[i])==0,則f[i]到i之間的部分及為這個串的最短迴圈節,具體畫個圖,或者參考白書上的圖,很容易就能看出來。

      另外這題字尾陣列應該也能寫,不過10^6的資料必須要寫DC3了,倍增的SA目測要T...

      上一個KMP的程式碼

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <queue>
#include <map>
#include <string>
#include <cstring>
#include <string>
using namespace std;
typedef long long ll;
const int maxn=1000000+100;
char str[maxn],s[maxn],s1[maxn];
int f[maxn];
void getFail(char* P,int* f)
{
    int m=strlen(P);
    f[0]=0;
    f[1]=0;
    for (int i=1; i<m; i++)
    {
        int j=f[i];
        while(j && P[i]!=P[j]) j=f[j];
        f[i+1]=P[i]==P[j]?j+1:0;
    }
}
int find(char* T,char* P,int* f)
{
    int n=strlen(T);
    int m=strlen(P);
    getFail(P,f);
    int j=0;
    for (int i=0; i<n; i++)
    {
        while(j && P[j]!=T[i]) j=f[j];
        if (P[j]==T[i]) j++;
        if (j==m) return i-m+1;
    }
}


int main()
{
//    freopen("in.txt","r",stdin);
    int tt;
    while(~scanf("%s",s))
    {
        if (s[0]=='.') break;
        getFail(s,f);
        int len=strlen(s);
        if (len % (len-f[len])==0)
        {
            printf("%d\n",len/(len-f[len]));
        }
        else printf("1\n");
    }
    return 0;
}