codeforces535D:Tavas and Malekas(KMP)

In the first sample test all strings of form "ioioi?" where the question mark replaces arbitrary English letter satisfy.

Here |x| denotes the length of string x.

Please note that it‘s possible that there is no such string (answer is 0).

这道题的题意真的是理解了相当之久，一个半小时差不多有了，

首先输入n,m，说明原序列长度为n，在原序列中有m个与s相互匹配的串

然后输入子串s

最后m个数，分别带别了这m个与s匹配的子串的其实位置是在哪里

问，在确定了这些位置的字符之后，原串有多少种可能

这道题带我复习了一遍KMP的next数组，上一次做KMP都还是N年以前的事情了

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <math.h>
#include <algorithm>
using namespace std;
#define ls 2*i
#define rs 2*i+1
#define up(i,x,y) for(i=x;i<=y;i++)
#define down(i,x,y) for(i=x;i>=y;i--)
#define mem(a,x) memset(a,x,sizeof(a))
#define w(a) while(a)
#define LL long long
const double pi = acos(-1.0);
#define Len 1000006
#define mod 1000000007
const int INF = 0x3f3f3f3f;

LL n,m,next[Len],ans,a[Len],len;
char str[Len];
bool hsh[Len];

void get_next()
{
    LL k = -1,j = 0;
    next[0] = -1;
    w(j<len)
    {
        if(k == -1 || str[k] == str[j])
        {
            j++;
            k++;
            next[j] = k;
        }
        else
        {
            k = next[k];
        }
    }
}

void dfs(LL k)
{
    if(k==0) return;
    hsh[len-k+1] = true;
    dfs(next[k]);
}

LL solve(LL k)
{
    int i;
    if(k < 0) return 0;
    LL ret = 1;
    up(i,1,k)
    ret = (ret*26)%mod;
    return ret;
}

int main()
{
    LL i,j,k,flag;
    scanf("%I64d%I64d",&n,&m);
    {
        flag = 1;
        scanf("%s",str);
        len = strlen(str);
        get_next();//求出next数组
        dfs(next[len]);//从最后一个开始，对循环串进行标记
        ans = 1;
        if(m==0)//没有的话，每个位置都是26种可能
        {
            printf("%I64d\n",solve(n));

        }
        else
        {
            scanf("%I64d",&a[1]);
            ans = solve(a[1]-1);//先看a[1]位置之前又几个，每一个都是26种可能
            up(i,2,m)
            {
                scanf("%I64d",&a[i]);
                if(a[i]-a[i-1]<len)//如果前后相隔不大于str，那么我们必须要看是否嵌入了循环体
                {
                    if(!hsh[a[i]-a[i-1]+1])
                    {
                        flag = 0;
                        printf("0\n");
                        break;
                    }
                }
                else//相隔大于len，那么出去str串中间多余的那几个字符每个都是26种可能
                    ans = (ans*solve(a[i]-a[i-1]-len))%mod;
            }
            if(flag)//最后几个字符每个也是26种可能
                printf("%I64d\n",ans*solve(n-a[m]-len+1)%mod);
        }
    }

    return 0;
}