[NOI2014]动物园
思路:
一开始先写了一个裸的KMP,在求next数组(f数组)的时候加了一个判断j<<1是否>i+1(若大于则说明有重叠部分),后来发现并不能这样做。因为这样会影响到后面next数组的求值。后来看了一个题解,里面的思路是重新做一遍求next的过程,同时再判断是否超过一半。
参考:http://blog.csdn.net/zhaoyh2000/article/details/54866734
1 #define maxl 1000000+1 2 #define mod 1000000007 3 #include<string> 4 #include<iostream> 5 using namespace std; 6 int n,f[maxl],num[maxl]; 7 string s; 8 int main() { 9 ios::sync_with_stdio(false); 10 cin>>n; 11 while(n--) { 12 cin>>s; 13 f[0]=-1; 14 f[1]=0; 15 num[0]=0; 16 num[1]=1; 17 int j=0; 18 unsigned long long ans=1ull; 19 for(int i=1;i<(int)s.length();i++) { 20 while(j>=0&&s[i]!=s[j]) j=f[j]; 21 f[i+1]=++j; 22 num[i+1]=num[j]+1; 23 } 24 j=0; 25 for(int i=1;i<(int)s.length();i++) { 26 while(j>=0&&s[i]!=s[j]) j=f[j]; 27 j++; 28 while((j<<1)>(i+1)) j=f[j]; 29 ans*=(unsigned long long)num[j]+1; 30 ans%=mod; 31 } 32 cout<<ans<<endl; 33 } 34 return 0; 35 }