2017多校联合第四场/HDU 6068 Classic Quotation(kmp+dp)

来源:互联网 发布:蚁群算法代码 编辑:程序博客网 时间:2024/05/17 03:42

Description 
给出一个长度为n的字符串S和一个长度为m的字符串T,有q次查询,每次查询给出一个区间[L,R],求对所有1<=i<=L,R<=j<=n,去掉S串的[i,j]部分后重组的串可以匹配T的次数之和 
Input 
第一行一整数T表示用例组数,每组用例首先输入三个整数n,m,q分别表示S串串长和T串串长以及查询次数,之后输入串S和串T,两个串均由小写字母构成,最后q行每行输入两个整数L,R表示一次查询 
(1<=T<=15,1<=n<=5e5,1<=m<=100,1<=q<=1e5,1<=L < R<=n) 
Output 
对于每次查询,输出查询结果 
Sample Input 

8 5 4 
iamnotsb 
iamsb 
4 7 
3 7 
3 8 
2 7 
Sample Output 




Solution 
这里写图片描述 
Code

#include<cstdio>using namespace std;namespace fastIO {    #define BUF_SIZE 100000    //fread -> read    bool IOerror=0;    inline char nc()     {        static char buf[BUF_SIZE],*p1=buf+BUF_SIZE,*pend=buf+BUF_SIZE;        if(p1==pend)         {            p1=buf;            pend=buf+fread(buf,1,BUF_SIZE,stdin);            if(pend==p1)             {                IOerror=1;                return -1;            }        }        return *p1++;    }    inline bool blank(char ch)     {        return ch==' '||ch=='\n'||ch=='\r'||ch=='\t';    }    inline void read(int &x)     {        char ch;        while(blank(ch=nc()));        if(IOerror)return;        for(x=ch-'0';(ch=nc())>='0'&&ch<='9';x=x*10+ch-'0');    }    inline void readc(char &x)    {        char ch;        while(blank(ch=nc()));        if(IOerror)return;        x=ch;    }    #undef BUF_SIZE};using namespace fastIO;typedef long long ll;#define maxn 50005#define maxm 105 int T,n,m,q;char a[maxn],b[maxm];int nxt[maxm];//b序列的next数组 int pre[maxn];//pre[i]表示a[1~i]中b匹配了多少次int num[maxn][maxm];//num[i][j]表示b与a的前i个前缀匹配的失配指针为j的数量int flag[maxm][27];//flag[i][j]表示b和a失配位置在i且下一次匹配的字母为j时是否可以完全匹配int nxt2[maxm][27];//nxt2[i][j]表示b和a失配位置在i且然后匹配j字母后的失配位置 int suf[maxn][maxm];//suf[i][j]表示当b和a的前缀失配位置在j时接着匹配a[i~n]可以匹配的数量int main(){    read(T);    while(T--)    {        read(n),read(m),read(q);        for(int i=1;i<=n;i++)readc(a[i]);        for(int i=1;i<=m;i++)readc(b[i]);        nxt[1]=0;        for(int j=0,i=2;i<=m;i++)        {            while(j&&b[j+1]!=b[i])j=nxt[j];            if(b[j+1]==b[i])j++;            nxt[i]=j;        }        for(int j=0,i=1;i<=n;i++)        {            while(j&&b[j+1]!=a[i])j=nxt[j];            if(b[j+1]==a[i])j++;            pre[i]=pre[i-1];            if(j==m)pre[i]++,j=nxt[j];            for(int k=0;k<m;k++)num[i][k]=num[i-1][k];            num[i][j]++;        }        for(int i=1;i<=n;i++)pre[i]+=pre[i-1];         for(int i=0;i<m;i++)            for(int j=0;j<26;j++)            {                flag[i][j]=0;                 int k=i;                while(k&&b[k+1]!=j+'a')k=nxt[k];                if(b[k+1]==j+'a')k++;                if(k==m)k=nxt[k],flag[i][j]=1;                nxt2[i][j]=k;            }        for(int i=0;i<26;i++)suf[n+1][i]=0;        for(int i=n;i>=1;i--)            for(int j=0;j<m;j++)                suf[i][j]=flag[j][a[i]-'a']+suf[i+1][nxt2[j][a[i]-'a']];        for(int i=n;i>=1;i--)            for(int j=0;j<m;j++)                suf[i][j]+=suf[i+1][j];        while(q--)        {            int l,r;            read(l),read(r);            ll ans=(ll)(n-r+1)*pre[l];            for(int i=0;i<m;i++)ans+=(ll)num[l][i]*suf[r][i];            printf("%I64d\n",ans);        }    }    return 0;}


原创粉丝点击