HDU 6068 Classic Quotation(kmp+dp)

xiaoxiao2021-02-28  87

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 1 8 5 4 iamnotsb iamsb 4 7 3 7 3 8 2 7 Sample Output 1 1 0 0 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; }
转载请注明原文地址: https://www.6miu.com/read-19300.html

最新回复(0)