[BZOJ]4974: 字符串大师 KMP

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。

题解:

对于 i!=a[i] ,直接 mod 一下就好了;对于 i=a[i] ,模仿KMP的 next 数组构造,一位一位跳,把不可行的字母标记一下,最后选个最小的。

代码:

#include
#include
#include
#include
using namespace std;
const int maxn=100010;
int n,a[maxn],len=1;
char ans[maxn];
int h[28],hh=0;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    ans[1]='a';
    for(int i=2;i<=n;i++)
    {
        if(a[i]!=i)ans[i]=ans[(i-1)%a[i]+1];
        else
        {
            hh++;
            int last=i-1;
            while(a[last]!=last)
            {
                last=(last-1)%a[last]+1;
                h[ans[last+1]-'a']=hh;
            }
            for(int j=1;j<26;j++)
            if(h[j]!=hh){ans[i]=j+'a';break;}
        }
    }
    for(int i=1;i<=n;i++)printf("%c",ans[i]);
}

你可能感兴趣的:(KMP)