Codeforces 535D - Tavas and Malekas【KMP】

题意：已知子串p是长度为n字符串的子串。已知p[0]出现的m个位置。问原字符串可能有多少种情况。

数据分析：1 ≤ n ≤ 10⁶ and 0 ≤ m ≤ n - |p| + 1

思路：把位置全填了就好了。但是直接填的复杂度到1e12会T。那么用KMP优化一下有冲突的位置

1.求next  得到 vis1

2.根据a[i]与a[i-1]的关系，判断如何填，能不能填

3.计算没填过的位置数量

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn=1e6+10000;
const int MOD=1e9+7;
int n,m;
int  a[maxn];
int  Next[maxn]; //不能用next
char s[maxn];
bool vis[maxn]; //原串填过的位置
bool vis1[maxn]; //原串那些位置会冲突
int len;

ll mod_pow(ll x,ll n)//快速幂
{
    ll res=1;
    while(n)
    {
        if(n%2==1)  res=res*x%MOD;
        x=x*x%MOD;
        n/=2;
    }
    return res;
}

void getnext()  //得到next
{
    memset(vis,false,sizeof(vis));
    memset(vis1,false,sizeof(vis1));
    int i=0,j=-1;
    Next[0]=-1;
    len=strlen(s);
    while(i<len)
    {
        if(j==-1 || s[i]==s[j])
            i++,j++,Next[i]=j;
        else    j=Next[j];
    }
    /*for(int i=0;i<=len;i++)
    {
        printf("next[%d]=%d\n",i,next[i]);
    }*/
    int r=len;
    while(r!=-1) //得到vis1
    {
        vis1[r]=true;
        r=Next[r];
    }
    /*for(int i=0;i<=len;i++)
    {
        printf("vis1[%d]=%d\n",i,vis1[i]);
    }*/
    return ;
}

int main(void)
{
    cin >> n >> m;
    scanf("%s",s);
    for(int i=1; i<=m; i++)   scanf("%d",&a[i]);
    getnext();
    sort(a+1,a+1+m);
    for(int i=1; i<=m; i++)
    {
        if(i==1 || a[i]-a[i-1]>=len) //如果是第一个或者相邻的位置长度可以填下一个len
        {
            for(int j=a[i];j<=a[i]+len-1;j++)
                vis[j]=true;
        }
        else//否则，如果冲突和不冲突
        {
            if(vis1[  a[i-1]+len-a[i] ])
            {
                for(int j=a[i-1]+len;j<=a[i]+len-1;j++)
                    vis[j]=true;
            }
            else
            {
                //cout << a[i-1] <<endl;
                printf("0\n");return 0;
            }
        }
    }
    ll ans=1;
    ll cnt=0;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]==false)
        {
            cnt++;
        }
    }
    ans=mod_pow(1LL*26,cnt);
    cout << ans << endl;
}