思路:这题确实大帝做得很机智!字符串先求最长前缀,反的字符串再求一次最长前缀,然后就可以搞了。
每个子串出现的次数就是最长前缀的次数嘛!
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<bitset>
#define mem(a,b) memset(a,b,sizeof(a))
#define lson i<<1,l,mid
#define rson i<<1|1,mid+1,r
#define llson j<<1,l,mid
#define rrson j<<1|1,mid+1,r
#define INF 0x7fffffff
#define maxn 100005
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
void EKMP(char *s,char *t,ll *next,ll *extend)//s[]为主串,t[]为模版串
{
int i,j,p,l;
int len=strlen(t);
int len1=strlen(s);
memset(next,0,sizeof(next));
memset(extend,0,sizeof(extend));
next[0]=len;
j=0;
while(1+j<len&&t[j]==t[1+j])j++;
next[1]=j;
int a=1;
for(i=2; i<len; i++)
{
p=next[a]+a-1;
l=next[i-a];
if(i+l<p+1)next[i]=l;
else
{
j=max(0,p-i+1);
while(i+j<len&&t[i+j]==t[0+j])j++;
next[i]=j;
a=i;
}
}
j=0;
while(j<len1&&j<len&&s[j]==t[j])j++;
extend[0]=j;
a=0;
for(i=1; i<len1; i++)
{
p=extend[a]+a-1;
l=next[i-a];
if(l+i<p+1)extend[i]=next[i-a];
else
{
j=max(0,p-i+1);
while(i+j<len1&&j<len&&s[i+j]==t[j])j++;
extend[i]=j;
a=i;
}
}
}
char S[maxn],T[maxn];
ll next[maxn],extend[maxn],sum[maxn],sum1[maxn];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
mem(sum,0);mem(sum1,0);
scanf("%s%s",S,T);
EKMP(S,T,next,extend);
int len1=strlen(S),len2=strlen(T);
for(int i=0;i<len1;i++)
sum[extend[i]]++;
for(int i=len2-1;i>=1;i--)
sum[i]+=sum[i+1];
reverse(S,S+len1);
reverse(T,T+len2);
EKMP(S,T,next,extend);
for(int i=0;i<len1;i++)
sum1[extend[i]]++;
for(int i=len2-1;i>=1;i--)
sum1[i]+=sum1[i+1];
ll ans=0;
for(int i=1;i<len2;i++)
ans+=sum[i]*sum1[len2-i];
printf("%lld\n",ans);
}
return 0;
}
/*
2
aaabbb
ab
ababaabaaaab
abaab
*/