【2017中国大学生程序设计竞赛 - 网络选拔赛】A Secret

【链接】http://acm.hdu.edu.cn/showproblem.php?pid=6153

【题意】

给出两个字符串S1,S2，求S2的所有后缀在S1中出现的次数与其长度的乘积之和。 

【题解】

扩展KMP的模板题.

首先,把S2和S1都倒转一下.

这样就转化成球S2的所有前缀在S1中出现的次数了.

而扩展KMP算法可以求出S1的每个位置i它的后缀i..lens1和S2[0..lens2-1]的最长公共前缀;

因为每个起点都不一样,则不会出现重复计算。

又因为S2的开头字符一定要涉及到;

所以这样是正确的。

根据这个extend[i]可以很容易搞出答案的.

【错的次数】

1

【反思】

(中间乘的时候会爆long long)

【代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define ms(x,y) memset(x,y,sizeof x)
#define ri(x) scanf("%d",&x)
#define rl(x) scanf("%lld",&x)
#define rs(x) scanf("%s",x)
#define oi(x) printf("%d",x)
#define ol(x) printf("%lld",x)
#define oc putchar(' ')
#define os(x) printf(x)
#define all(x) x.begin(),x.end()
#define Open() freopen("F:\\rush.txt","r",stdin)
#define Close() ios::sync_with_stdio(0)
 
typedef pair<int,int> pii;
typedef pair<LL,LL> pll;
 
const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);
 
const int N=1e6;
const LL MOD = 1e9+7;
int Next[N+10],extend[N+10],lent,lens;
char S[N+10],T[N+10];
 
void makenext(int m){
    int a = 0;
    Next[0] = lens;
    while(a < lens - 1 && S[a] == S[a + 1]) a++;
    Next[1] = a;
    a = 1;
 
    for(int k = 2; k < lens; k ++) {
        int p = a + Next[a] - 1,L = Next[k - a];
        if( (k - 1) + L >= p) {
            int j = (p - k + 1) > 0 ? (p - k + 1) : 0;
            while(k + j < lens && S[k + j] == S[j]) j++;
            Next[k] = j;
            a = k;
        } else
            Next[k] = L;
    }
}
void GetNext(const char *T){
     int a=0;
     int MinLen = lens < lent ? lens : lent;
     while(a < MinLen && S[a] == T[a] ) a++;
     extend[0]=a;
     a=0;
     for(int k=1;k < lent;k++){
         int p=a+extend[a]-1,L = Next[k-a];
         if((k-1)+L>=p){
             int j=(p-k+1)>0? (p-k+1):0;
             while(k + j < lent && T[k+j] == S[j]) j++;
             extend[k]=j;
             a=k;
         }
         else extend[k]=L;
     }
}
 
int main(){
    //Open();
    int TT;
    ri(TT);
    while (TT--){
        rs(T),rs(S);
        lent = strlen(T),lens = strlen(S);
        reverse(T,T+lent),reverse(S,S+lens);
        makenext(lens);
        GetNext(T);
        LL sum = 0;
        rep1(i,0,lent-1){
            //extend[i] = min(extend[i],lens);
            // 1 + 2 + .. extend[i]
            sum = (sum + 1LL*(1 + extend[i])*extend[i]/2)%MOD;
        }
        ol(sum);puts("");
    }
    return 0;
}