codeforces 536B Tavas and Malekas - Z-Box算法

时间:2022-11-09 06:01:25

传送门
题解:模板题

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define mod 1000000007
#define N 1000010
#define lint long long
using namespace std;
char s[N];int z[N],n,x[N],cnt[N];
inline bool can(int x,int y)
{
    if(y>=x+n) return true;
    return z[y-x+1]>=x+n-y;
}
int fast_pow(int x,int k)
{
    if(!k) return 1;
    if(k==1) return x%mod;
    int ans=fast_pow(x,k>>1);
    ans=(lint)ans*ans%mod;
    if(k&1) ans=(lint)ans*x%mod;
    return ans;
}
int main()
{
    int m,q;scanf("%d%d",&m,&q);
    if(!q) return !printf("%d\n",fast_pow(26,m));
    scanf("%s",s+1);n=strlen(s+1);
    for(int i=2,l=1,r=0;i<=n;i++)
    {
        (i>r)?(z[i]=(s[i]==s[1])):(z[i]=min(z[r-l+1-(r-i)],r-i+1));
        while(i+z[i]<=n&&s[i+z[i]]==s[1+z[i]]) z[i]++;
        if(i+z[i]>r&&z[i]) l=i,r=i+z[i]-1;
    }
    for(int i=1;i<=q;i++) scanf("%d",&x[i]);
    for(int i=1;i<q;i++)
        if(can(x[i],x[i+1]))
            cnt[x[i]]++,cnt[x[i]+n]--;
        else return !printf("0\n");
    cnt[x[q]]++,cnt[x[q]+n]--;int c=0;
    for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for(int i=1;i<=m;i++) if(!cnt[i]) c++;
    printf("%d\n",fast_pow(26,c));
    return 0;
}