【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\dots lens-1]$与$p[0\dots lenp-1]$的最长公共前缀长度
比如翻转后$s = babababa$,$p = aba$
$ex[1] = 3$.
可以说明$a$,$ab$,$aba$都至少出现了一次,对答案的贡献是$1+2+3$,也就是$ex[i] * (ex[i]+1) / 2$

实际上这里$a$在$s[1\dots 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;
}