标签:hdu his logs ios i++ hat include 题目 数组
给你两个字符串A,B,现在要你求B串的后缀在A串中出现的次数和后缀长度的乘积和为多少。
扩展KMP模板题,将s和t串都逆序以后就变成了求前缀的问题了,扩展KMP求处从i位置开始的最长公共前缀存于数组,最后通过将数组的值不为0的进行一个等差数列和的和就可以了。
代码:
1 #include <iostream> 2 #include <string> 3 #include <string.h> 4 #include <cstring> 5 #include <algorithm> 6 using namespace std; 7 const int maxn = 1e6 + 10; 8 const int mod = 1e9 + 7; 9 typedef long long ll; 10 int cnt[maxn]; 11 char A[maxn],B[maxn]; 12 int Next[maxn],ex[maxn]; 13 ll add(ll n) 14 { 15 ll m=((n%mod)*((n+1)%mod)/2)%mod; 16 return m; 17 } 18 void kmp(char P[]) 19 { 20 int m=strlen(P); 21 Next[0]=m; 22 int j=0,k=1; 23 while(j+1<m&&P[j]==P[j+1]) j++; 24 Next[1]=j; 25 for(int i=2; i<m; i++) 26 { 27 int p=Next[k]+k-1; 28 int L=Next[i-k]; 29 if(i+L<p+1) Next[i]=L; 30 else 31 { 32 j=max(0,p-i+1); 33 while(i+j<m&&P[i+j]==P[j]) 34 j++; 35 Next[i]=j; 36 k=i; 37 } 38 } 39 } 40 41 void exkmp(char P[],char T[]) 42 { 43 int m=strlen(P),n=strlen(T); 44 kmp(P); 45 int j=0,k=0; 46 while(j<n&&j<m&&P[j]==T[j]) 47 j++; 48 ex[0]=j; 49 for(int i=1; i<n; i++) 50 { 51 int p=ex[k]+k-1; 52 int L=Next[i-k]; 53 if(i+L<p+1) 54 ex[i]=L; 55 else 56 { 57 j=max(0,p-i+1); 58 while(i+j<n&&j<m&&T[i+j]==P[j]) 59 j++; 60 ex[i]=j; 61 k=i; 62 } 63 } 64 } 65 66 int main() 67 { 68 int t; 69 scanf("%d",&t); 70 while(t--) 71 { 72 scanf("%s%s",A,B); 73 int lenA=strlen(A); 74 int lenB=strlen(B); 75 reverse(A,A+lenA); 76 reverse(B,B+lenB); 77 kmp(B); 78 memset(Next,0,sizeof(Next)); 79 memset(ex,0,sizeof(ex)); 80 exkmp(B,A); 81 ll ans = 0; 82 for(int i=0;i<lenA;i++) 83 { 84 if(ex[i]) 85 ans=(ans+add(ex[i])%mod)%mod; 86 } 87 printf("%lld\n",ans%mod); 88 } 89 return 0; 90 }
标签:hdu his logs ios i++ hat include 题目 数组
原文地址:http://www.cnblogs.com/mj-liylho/p/7400064.html