A - A Secret （扩展kmp）

题目链接：https://cn.vjudge.net/contest/283743#problem/A

题目大意：给你字符串s1和s2，然后问你s2的每一个后缀在s1中出现的次数之和（可重叠）。

具体思路：首先将s1和s2翻转过来，这样的话就把后缀问题转换成了求前缀的问题。举个例子，s1="abcd"，s2="cd"，将两个字符串都翻转之后，s1变成了"dcba",s2变成了"dc"，然后就按照扩展kmp的方法求出s2的每一个后缀在s1中的出现的位置，样例二，s1="abababab",  s2="aba".翻转之后s1="babababa",s2="aba"。然后扩展kmp求出extend数组之后， 0 3 0 3 0 3 0 1，然后3代表的s2的前3个字符是和s1相同的，那么前两个也是相同的，前一个也是相同的，所以打个桶记录前缀和就能表示出所有的满足情况了。

AC代码：

 #include<iostream>
 #include<stack>
 #include<iomanip>
 #include<stdio.h>
 #include<cmath>
 #include<algorithm>
 #include<string>
 #include<map>
 #include<cstring>
 #include<vector>
 using namespace std;
 # define ll long long
 const int mod = 1e9+;
 const int maxn = 1e6+;
 char str1[maxn];
 char str2[maxn];
 int nex[maxn],extend[maxn];
 ll vis[maxn];
// 注意vis开long long ，否则会爆精度。
// str2是需要匹配的串，str1是被匹配的串。
 void getnex(int len)
 {
     int a=,p=;
     nex[]=len;
     for(int i=; i<len; i++)
     {
         if(i>=p||i+nex[i-a]>=p)
         {
             if(i>=p)
                 p=i;
             while(p<len&&str2[p]==str2[p-i])
                 p++;
             nex[i]=p-i;
             a=i;
         }
         else
             nex[i]=nex[i-a];
     }
 }
 void exkmp(int len1,int len2)
 {
     getnex(len2);
     int a=,p=;
     for(int i=; i<len1; i++)
     {
         if(i>=p||i+nex[i-a]>=p)
         {
             if(i>=p)
                 p=i;
             while(p<len1&&p-i<len2&&str1[p]==str2[p-i])
                 p++;
             extend[i]=p-i;
             a=i;
         }
         else
             extend[i]=nex[i-a];
     }
 }
 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         memset(vis,,sizeof(vis));
         scanf("%s",str1);
         scanf("%s",str2);
         int len1=strlen(str1);
         int len2=strlen(str2);
         reverse(str1,str1+len1);
         reverse(str2,str2+len2);
         exkmp(len1,len2);
         for(int i=;i<len1;i++){
         vis[extend[i]]++;
         }
         ll ans=len2*vis[len2]%mod;
         for(int i=len2;i>=;i--){
         vis[i-]+=vis[i];
         ans=(ans+vis[i-]*(i-)%mod+mod)%mod;
         }
         printf("%lld\n",ans);
     }
     return ;
 }