HDU 6068 Classic Quotation（kmp+dp）

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;
}