[POJ](2406)Power Strings ---- KMP（连续重复子串）

题目传送门

做法：

• KMP利用最小循环节周期
• 那么 周期个数 = len/ len - nex[len] 即是答案
• 看了罗神的论文，试着用后缀数组做一下，TLE，这个题倍增会被卡常，DC3可过，但是时间复杂度远大于KMP。
• 总结：
• KMP为最优解
• 有时候同一个题，从不同的做法切入，即是A不掉，也是一次不错的收获(＾－＾)V
• 温故知新~

AC代码：

//#include<bits/stdc++.h>
#include <iostream>
#include <string>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <algorithm>
#define IO          ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define pb(x)       push_back(x)
#define sz(x)       (int)(x).size()
#define sc(x)       scanf("%d",&x)
#define abs(x)      ((x)<0 ? -(x) : x)
#define all(x)      x.begin(),x.end()
#define mk(x,y)     make_pair(x,y
#define fin         freopen("in.txt","r",stdin)
#define fout        freopen("out.txt","w",stdout)
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int mod = 1e9+7;
const double PI = 4*atan(1.0);
const int maxm = 1e5+5;
const int maxn = 1e6+5;
const int INF = 0x3f3f3f3f;
char s[maxn];
int nex[maxn],n;
void get_nex()
{
    int k = nex[0] = -1;
    int i = 0;
    while(i<n){
        if(k<0 || s[i] == s[k]){
            nex[++i] = ++k;
        }
        else k = nex[k];
    }
}
int main()
{
    // fin;
    while(scanf("%s",s))
    {
        memset(nex,0,sizeof(nex));
        if(s[0] == '.') break;
        n = strlen(s);
        get_nex();
        int ans = -1;
        int mnlen = n-nex[n];
        if(nex[n] == 0) ans = 1;
        else if(n%mnlen == 0) ans = n/mnlen;
        else if(n%mnlen != 0) ans = 1;
        printf("%d\n",ans);
    }
    return 0;
}

TLE代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define IO          ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define pb(x)       push_back(x)
#define sz(x)       (int)(x).size()
#define sc(x)       scanf("%d",&x)
#define abs(x)      ((x)<0 ? -(x) : x)
#define all(x)      x.begin(),x.end()
#define mk(x,y)     make_pair(x,y)
#define debug       printf("!!!!!!\n")
#define fin         freopen("in.txt","r",stdin)
#define fout        freopen("out.txt","w",stdout)
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int mod = 1e9+7;
const double PI = 4*atan(1.0);
const int maxm = 1e6;
const int maxn =1e6+10;
const int INF = 0x3f3f3f3f;
const ll LINF = 1ll<<62;
const int m = 128; //
char s[maxn]; //待排序的字符串放在s数组中,从s[0~n-1],长度为n,且最大值小于m,最后一位是0(无效值)
int sa[maxn],t1[maxn],t2[maxn],c[maxn],n; //sa[1~n]为有效值,sa[0]必定为n是无效值
int rk[maxn]; //rank[0~n-1]为有效值,rank[n]必定为0无效值
int height[maxn]; //height[1~n]
//lcp(x,y)：字符串x与字符串y的最长公共前缀，在这里指x号后缀与与y号后缀的最长公共前缀
//height[i]：lcp(sa[i],sa[i-1])，即排名为i的后缀与排名为i-1的后缀的最长公共前缀
//H[i]：height[rak[i]]，即i号后缀与它前一名的后缀的最长公共前缀
void build_sa(int m)
{
    n++;
    int *x = t1,*y = t2;
    for(int i=0;i<m;i++) c[i] = 0;
    for(int i=0;i<n;i++) c[x[i] = s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]] = i;
    for(int j=1;j<=n;j<<=1)
    {
        int p = 0;
        for(int i = n-j;i<n;i++) y[p++] = i;
        for(int i=0;i<n;i++) if(sa[i]>=j) y[p++] = sa[i]-j;
        for(int i=0;i<m;i++) c[i] = 0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        p = 1;x[sa[0]] = 0;
        for(int i=1;i<n;i++)
            x[sa[i]] = (y[sa[i-1]] == y[sa[i]] && y[sa[i-1]+j] == y[sa[i]+j])?p-1:p++;
        if(p>=n) break;
        m = p;
    }
    n--;
    int k = 0;
    for(int i=0;i<=n;i++) rk[sa[i]] = i;
    for(int i=0;i<n;i++){
        if(k) k--;
        int j = sa[rk[i]-1];
        while(s[i+k] == s[j+k]) k++;
        height[rk[i]] = k;
    }
}
int query(int i,int j)
{
    int l = rk[0];
    int r = rk[j];
    if(l>r) swap(l,r);
    l++;
    return min(height[l],height[r]);
}
int main()
{
    // fin;
    // IO;
    while(scanf("%s",s))
    {
        if(s[0] == '.') break;
        n = strlen(s);
        build_sa(m);
        int ans = -1;
        for(int i=1;i<=n;i++) //枚举长度
        {
            if(n%i == 0)
                if(query(0,i) == n-i) ans = max(ans,n/i);
        }
        printf("%d\n",ans);
    }
    return 0;
}