【HDU6153】A Secret(扩展KMP)

记录一个菜逼的成长。。

题目链接

2017CCPC网络赛

题目大意:

给你一个字符串s,一个字符串p.
问p的所有后缀字符串在s中出现的次数乘上相应后缀字符串的长度的和

Hint:

sample input:
abababab
aba
Suffix(S2,1) = “aba”,
Suffix(S2,2) = “ba”,
Suffix(S2,3) = “a”.
N1 = 3,
N2 = 3,
N3 = 4.
L1 = 3,
L2 = 2,
L3 = 1.
ans = （3*3+3*2+4*1）%1000000007.

笔记

先将两个字符串都reverse一下,那么第二字符串的后缀现在变成了前缀。
跑一遍扩展KMP
求出ex[i]数组 := 表示s[i…lens−1]与p[0…lenp−1]的最长公共前缀长度
比如翻转后s=babababa,p=aba
ex[1]=3.
可以说明a,ab,aba都至少出现了一次,对答案的贡献是1+2+3,也就是ex[i]∗(ex[i]+1)/2

实际上这里a在s[1…3]这里出现了两次，不要担心。在ex[3]=3时，会算上。所以是不会漏的。

ps:对字符串的套路知之甚少。。orz.为什么你们什么都会。

#include <bits/stdc++.h>using namespace std;#define cl(a,b) memset(a,b,sizeof(a))typedef long long LL;const int INF = 0x3f3f3f3f;const int MOD = 1e9 + 7;const int maxn = 1000000 + 10;char s[maxn],p[maxn];int _next[maxn],ex[maxn];/*** 扩展KMP算法**///next[i]:x[i...m-1]与x[0...m-1]的最长公共前缀//extend[i]:y[i...n-1]与x[0...m-1]的最长公共前缀void getNext(char x[],int m,int next[]){  _next[0]=m;  int j=0;  while(j+1 < m && x[j] == x[j+1])j++;  _next[1] = j;  int k = 1;  for (int i = 2; i < m; i++ ) {    int p = _next[k] + k - 1;    int L = _next[i-k];    if(i + L < p + 1) _next[i] = L;    else {      j = max(0,p-i+1);      while(i+j < m && x[i+j] == x[j]) j++;      _next[i] = j;      k = i;    }  }}void EKMP(char x[],int m,char y[],int n,int next[],int extend[]){  getNext(x,m,next);  int j = 0;  while(j < n && j < m && x[j] == y[j])j++;  extend[0] = j;  int k = 0;  for (int i = 1; i < n; i++) {    int p = extend[k] + k - 1;    int L = next[i-k];    if(i + L < p + 1)extend[i] = L;    else {      j = max(0,p-i+1);      while(i + j < n && j < m && y[i+j] == x[j]) j++;      extend[i]=j;      k = i;    }  }}int main(){  int T;scanf("%d",&T);  while(T--){    scanf("%s%s",s,p);    int lens = strlen(s);    int lenp = strlen(p);    reverse(s,s+lens);    reverse(p,p+lenp);    EKMP(p,lenp,s,lens,_next,ex);    LL ans = 0;    for (int i = 0; i < lens; i++) {      ans = (ans + (LL)ex[i] * (ex[i]+1) % MOD * 500000004LL % MOD) % MOD;    }    printf("%lld\n",ans % MOD);  }  return 0;}