BZOJ4974字符串大师——KMP

个人BZOJ第150题，撒花~~~~~

---

Description

一个串T是S的循环节，当且仅当存在正整数k，使得S是T^k(即T重复k次)的前缀，比如abcd是abcdabcdab的循环节
。给定一个长度为n的仅由小写字符构成的字符串S，请对于每个k(1<=k<=n)，求出S长度为k的前缀的最短循环节的
长度per_i。字符串大师小Q觉得这个问题过于简单，于是花了一分钟将其AC了，他想检验你是否也是字符串大师。
小Q告诉你n以及per_1,per_2,…,per_n，请找到一个长度为n的小写字符串S，使得S能对应上per。
Input

第一行包含一个正整数n(1<=n<=100000)，表示字符串的长度。
第二行包含n个正整数per_1,per_2,…per_n(1<=per_i<=i)，表示每个前缀的最短循环节长度。
输入数据保证至少存在一组可行解。
Output

输出一行一个长度为n的小写字符串S，即某个满足条件的S。
若有多个可行的S，输出字典序最小的那一个。
Sample Input

5

1 2 2 2 5
Sample Output

ababb

---

我们分两种情况讨论，第一种per_i<i,也就是循环节已经在之前出现过了。那么这种情况是很简单的，因为当前的点的循环节是在之前出现过的，且大小为per_i,那么从0~per_i-1，就是第一个循环节，那我们只要用i%per_i,找出其在第一个循环节上的位置即可。

        if(a[i]<i){
            if(i%a[i]==0) s[i-1]=s[a[i]-1];
            else s[i-1]=s[i%a[i]-1];
        }

关键是per_i==i的情况，那也就是说当前的串的最小循环节就是它本身，那么我们考虑小于它本身的情况，假设一个字符串是由一个循环节构成的，那么构成的模式图肯定如下图：

每一个绿色的框就是一个循环节，而红色是多出来的一段，而涂成棕色的部分是与红色相同的一小段。那么我们发现用蓝色框框起来的两段区间，它们中的字符串是一样的，并且大于1。

所以我们可以对当前整个字符串求next数组，如果next[i]>1，那么说明当前存在更小的循环节。因此我们只要从a到z枚举哪一位放入当前位置，然后判断现在是否拥有更小的循环节，如果没有，就放入这个字符，然后处理下一位即可。

由于求next数组复杂度是O(n)的，又因为字符串之前的部分不会变，所以每次求next并不用从头开始，只要从上次的位置即可。当然枚举完每一位后还是要往前回溯两位。所以这样最坏的平均复杂度也是O(n)加26的常数。

        else{
            s[i-1]='a'+now;
            if(i==2) s[i-1]++;
            else while(check(i)){
                l-=2,r=nxt[l],s[i-1]++;
            } 
        }

完整代码：

#include<bits/stdc++.h>
using namespace std;
int read(){
    char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
int n,now,l,r,lst,a[100005],nxt[100005];
char s[100005];
void getnext(int x){
    while(l<x&&r<x){
        if(r==-1||s[l]==s[r]){
            l++;r++;nxt[l]=r;
        }
        else r=nxt[r];
    }
}
int check(int x){
    getnext(x);
    if(nxt[x]<=1&&s[x-1]!=s[0]) return 0;  //这里还得特判s[x-1]!=s[0]，因为若真实的循环节为x-1，那么next会是1
    else return 1;
}
int main()
{
    n=read();s[0]='a';lst=0;nxt[0]=-1;l=0;r=-1;
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=2;i<=n;i++){
        if(a[i]<i){
            if(i%a[i]==0) s[i-1]=s[a[i]-1];
            else s[i-1]=s[i%a[i]-1];
        }
        else{
            s[i-1]='a'+now;
            if(i==2) s[i-1]++;
            else while(check(i)){
                l-=2,r=nxt[l],s[i-1]++;
            } 
        }
    }
    cout<<s;
    return 0;   
}