标签:
Time Limit: 20 Sec
Memory Limit: 256 MB
http://uoj.ac/problem/5
Input
Output
输出文件应包含n行
每行描述一组测试数据的答案
答案的顺序应与输入数据的顺序保持一致。对于每组测试数据,仅需要输出一个整数,表示这组测试数据的答案对1000000007取模的结果。
输出文件中不应包含多余的空行
Sample Input
3
aaaaa
ab
abcababc
Sample Output
36
1
32
n,m≤105 q≤105
题意
题解:
kmp算法的灵活运用
考察的算法是KMP的next数组的灵活运用和倍增思想。
因为next数组的定义是某个前缀的 前缀的后缀,也是s的前缀。满足这一条件的 前缀的后缀 的最长长度是多少。
如果没有“重叠”的限制的话,那么原问题就相当于求沿next指针上跳的层数,dp即可。
如果加上了重叠的限制呢?
那么就要沿着next指针上跳,找到第一个位置j,满足j + j < i(这里的编号是从0到n-1),那么往上跳的所有层数都是满足条件的。
如何找到这个位置?
显然暴力的跳会超时,那么我们可以参照倍增思想,记f[i][j]为从位置j沿next指针往上跳2^i次到达的位置,沿f数组上溯即可。
代码:
#include <cstdio> #include <cmath> #include <cstring> #include <ctime> #include <iostream> #include <algorithm> #include <set> #include <vector> #include <sstream> #include <queue> #include <typeinfo> #include <fstream> #include <map> #include <stack> typedef long long ll; using namespace std; //freopen("D.in","r",stdin); //freopen("D.out","w",stdout); #define sspeed ios_base::sync_with_stdio(0);cin.tie(0) #define test freopen("test.txt","r",stdin) #define maxn 1000010 #define mod 1000000007 #define eps 1e-6 const int inf=0x3f3f3f3f; const ll infll = 0x3f3f3f3f3f3f3f3fLL; inline ll read() { ll x=0,f=1;char ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();} return x*f; } //************************************************************************************** ll ans=1; char s[maxn]; int nex[maxn],fail[maxn]; void kmp() { int n=strlen(s+1); int j=0; nex[1]=1; for(int i=2;i<=n;i++) { while(s[j+1]!=s[i]&&j) j=fail[j]; if(s[j+1]==s[i]) j++; fail[i]=j; nex[i]=nex[j]+1; } j=0; for(int i=2;i<=n;i++) { while(s[j+1]!=s[i]&&j) j=fail[j]; if(s[j+1]==s[i]) j++; while((j<<1)>i&&j) j=fail[j]; ans=(ans*(nex[j]+1))%mod; } } int main() { int t=read(); while(t--) { ans=1; scanf("%s",s+1); kmp(); int n=strlen(s+1); for(int i=1;i<=n;i++) cout<<nex[i]<<" "; cout<<endl; for(int i=1;i<=n;i++) cout<<fail[i]<<" "; cout<<endl; cout<<ans<<endl; } }
标签:
原文地址:http://www.cnblogs.com/qscqesze/p/4585333.html