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