A - A Secret -扩展KMP
生活随笔
收集整理的這篇文章主要介紹了
A - A Secret -扩展KMP
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目大意: 給你兩個字符串A,B,現在要你求B串的后綴在A串中出現的次數和后綴長度的乘積和為多少。 題解: 擴展KMP模板題,將A和B串都逆序以后就變成了求前綴的問題了,擴展KMP求處從i位置開始的最長公共前綴存于數組。 最后通過將數組的值不為0的進行一個桶計數,倒著進行一下求和就可以了。注意,在這個題目上擴展kmp處理出來的是 ex[ i ]數組是 A串的每個從 i 位置開始的后綴 ,與B串的最長公共前綴長度,那么這樣B串在A串上匹配的情況就一目了然了。 #include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int N = 1000005;
int Next[N];
long long ex[N],tong[N]; //即extand[]
char p[N],t[N];
int T;
long long ans;
void pre() // next[i]: 以第i位置開始的子串 與 T的公共前綴
{int lp=strlen(p);Next[0]=lp;int j=0,k=1;while(j+1<lp && p[j]==p[j+1]) j++;Next[1]=j;for(int i=2; i<lp; 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<lp && p[i+j]==p[j]) j++; // 枚舉(p+1,length) 與(p-k+1,length) 區間比較Next[i]=j;k=i;}}
}
void exkmp()
{int lp=strlen(p),lt=strlen(t);pre(); //next數組初始化int j=0,k=0;while(j<lt && j<lp && p[j]==t[j]) j++;ex[0]=j;for(int i=1; i<lt; i++){int P=ex[k]+k-1;int L=Next[i-k];if(i+L<P+1) ex[i]=L;else{j=max(0,P-i+1);while(i+j<lt && j<lp && t[i+j]==p[j]) j++;ex[i]=j;k=i;}}
}
int main()
{scanf("%d",&T);while(T--){memset(tong,0,sizeof(tong));scanf("%s%s",&t,&p);int lt=strlen(t);int lp=strlen(p);reverse(p,p+lp);reverse(t,t+lt);exkmp();ans=0;for(int i=0; i<lt; i++)tong[ex[i]]++;for(int i=lp;i;i--){tong[i]=(tong[i]+tong[i+1])%mod;ans=(ans+tong[i]*i%mod)%mod;}printf("%lld\n",ans);}
}
轉載于:https://www.cnblogs.com/SDUTNING/p/10388792.html
總結
以上是生活随笔為你收集整理的A - A Secret -扩展KMP的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: JavaScript数据结构与算法——集
- 下一篇: sharding-jdbc学习