[luogu P2375] [NOI 2014] 动物园
题目描述
近日,园长发现动物园中好吃懒做的动物越来越多了。例如企鹅,只会卖萌向游客要吃的。为了整治动物园的不良风气,让动物们凭自己的真才实学向游客要吃的,园长决定开设算法班,让动物们学习算法。
某天,园长给动物们讲解KMP算法。
园长:“对于一个字符串S,它的长度为L。我们可以在O(L)的时间内,求出一个名为next的数组。有谁预习了next数组的含义吗?”
熊猫:“对于字符串S的前i个字符构成的子串,既是它的后缀又是它的前缀的字符串中(它本身除外),最长的长度记作next[i]。”
园长:“非常好!那你能举个例子吗?”
熊猫:“例S为abcababc,则next[5]=2。因为S的前5个字符为abcab,ab既是它的后缀又是它的前缀,并且找不到一个更长的字符串满足这个性质。同理,还可得出next[1] = next[2] = next[3] = 0,next[4] = next[6] = 1,next[7] = 2,next[8] = 3。”
园长表扬了认真预习的熊猫同学。随后,他详细讲解了如何在O(L)的时间内求出next数组。
下课前,园长提出了一个问题:“KMP算法只能求出next数组。我现在希望求出一个更强大num数组一一对于字符串S的前i个字符构成的子串,既是它的后缀同时又是它的前缀,并且该后缀与该前缀不重叠,将这种字符串的数量记作num[i]。例如S为aaaaa,则num[4] = 2。这是因为S的前4个字符为aaaa,其中a和aa都满足性质‘既是后缀又是前缀’,同时保证这个后缀与这个前缀不重叠。而aaa虽然满足性质‘既是后缀又是前缀’,但遗憾的是这个后缀与这个前缀重叠了,所以不能计算在内。同理,num[1] = 0,num[2] = num[3] = 1,num[5] = 2。”
最后,园长给出了奖励条件,第一个做对的同学奖励巧克力一盒。听了这句话,睡了一节课的企鹅立刻就醒过来了!但企鹅并不会做这道题,于是向参观动物园的你寻求帮助。你能否帮助企鹅写一个程序求出num数组呢?
特别地,为了避免大量的输出,你不需要输出num[i]分别是多少,你只需要输出所有num[i]的乘积,对1,000,000,007取模的结果即可。
输入输出格式
输入格式:
第1行仅包含一个正整数n ,表示测试数据的组数。随后n行,每行描述一组测试数据。每组测试数据仅含有一个字符串S,S的定义详见题目描述。数据保证S 中仅含小写字母。输入文件中不会包含多余的空行,行末不会存在多余的空格。
输出格式:
包含 n 行,每行描述一组测试数据的答案,答案的顺序应与输入数据的顺序保持一致。对于每组测试数据,仅需要输出一个整数,表示这组测试数据的答案对 1,000,000,007 取模的结果。输出文件中不应包含多余的空行。
输入输出样例
说明
测试点编号 约定
1 N ≤ 5, L ≤ 50
2 N ≤ 5, L ≤ 200
3 N ≤ 5, L ≤ 200
4 N ≤ 5, L ≤ 10,000
5 N ≤ 5, L ≤ 10,000
6 N ≤ 5, L ≤ 100,000
7 N ≤ 5, L ≤ 200,000
8 N ≤ 5, L ≤ 500,000
9 N ≤ 5, L ≤ 1,000,000
10 N ≤ 5, L ≤ 1,000,000
这题考的主要是对kmp的理解。
根据next数组的定义,在原串s里,对于每一个位置i,有s[1..next[i]]=s[i-next[i]+1..i]。
那么,对于前缀s[1..i]来说,num[i]一点和num[next[i]]有关。
我们先不考虑前后缀重叠的情况。那么,显然num[i]=num[next[i]]+1。
看这个图,由于我们的next[i]记录的都是最长的,所以不会担心遗漏掉某一个。
而由于s[1..next[next[i]]]=s[next[i]-next[next[i]]+1..next[i]]=s[i-next[next[i]]+1..i],
所以s[1..next[i]]是一种,s[1..next[next[i]]]是一种,s[1..next[next[next[i]]]]也是一种。(注意边界的初值)
那么,我们就可以说,dp[i]从dp[next[i]]转移过来,dp[next[i]]从dp[next[next[i]]]转移过来。。。(dp[0]的初值注意一下)
然后,题目中题意稍有不清,我也不太清楚它每个num实际都多了个1到底是哪里来的。
可能是空串也算一种吧。
然后,还有一个东西没讲呢。
就是万一next[j]>(i>>1)怎么办?缩小呗。看上面的图,一直往next[j]走,知道next[j]<=(i<<1)为止。
至于复杂度的证明,就是和证明kmp是O(n)的差不多,++j就n次,那减少的次数总共也不超过n次。(j=next[j]会导致j变小)
然后的话,上面那个特殊操作(就上2行)会不会影响后面的答案呢?这应当是一个贪心策略,在保证了不重叠的情况下取到最大,使得最终复杂度是O(n)的。
否则实测会T。
code:
View Code1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 typedef long long LL; 5 namespace fastIO { 6 #define puc(c) putchar(c) 7 inline int read() { 8 int x=0,f=1; char ch=getchar(); 9 while (ch<'0'||ch>'9') { 10 if (ch=='-') f=-f; 11 ch=getchar(); 12 } 13 while (ch>='0'&&ch<='9') { 14 x=(x<<3)+(x<<1)+ch-'0'; 15 ch=getchar(); 16 } 17 return x*f; 18 } 19 template <class T> inline void read(T &x=0) { 20 T f=1; char ch=getchar(); 21 while (ch<'0'||ch>'9') { 22 if (ch=='-') f=-f; 23 ch=getchar(); 24 } 25 while (ch>='0'&&ch<='9') { 26 x=(x<<3)+(x<<1)+ch-'0'; 27 ch=getchar(); 28 } 29 x*=f; 30 } 31 int cnt,w[20]; 32 template <class T> inline void write(T x) { 33 if (x==0) { 34 puc('0'); 35 return; 36 } 37 if (x<0) { 38 x=-x; 39 puc('-'); 40 } 41 for (cnt=0; x; x/=10) w[++cnt]=x%10; 42 for (; cnt; --cnt) puc(w[cnt]+48); 43 } 44 inline void newline() { 45 puc('\n'); 46 } 47 inline void newblank() { 48 puc(' '); 49 } 50 } 51 namespace OJ{ 52 void Online_Judge() { 53 #ifndef ONLINE_JUDGE 54 freopen("in.txt","r",stdin); 55 freopen("out.txt","w",stdout); 56 #endif 57 } 58 } 59 const int N=1000005; 60 const LL mo=1000000007; 61 char ori[N]; int n,mat[N]; LL ans,dp[N]; 62 void get_match() { 63 dp[0]=0,mat[1]=0,dp[1]=1; 64 for (int i=2,j=0; i<=n; ++i) { 65 while (j>0&&ori[j+1]!=ori[i]) j=mat[j]; 66 if (ori[j+1]==ori[i]) ++j; 67 mat[i]=j,dp[i]=dp[j]+1; 68 } 69 } 70 void KMP_match() { 71 ans=1; 72 for (int i=2,j=0; i<=n; ++i) { 73 while (j>0&&ori[j+1]!=ori[i]) j=mat[j]; 74 if (ori[j+1]==ori[i]) ++j; 75 while ((i>>1)<j) j=mat[j]; 76 ans=(ans*std::max(1LL,dp[j]+1))%mo; 77 } 78 fastIO::write(ans); 79 fastIO::newline(); 80 } 81 int main() { 82 OJ::Online_Judge(); 83 int T; scanf("%d",&T); 84 for (; T; --T) { 85 scanf("%s",ori+1),n=strlen(ori+1); 86 get_match(); 87 KMP_match(); 88 } 89 return 0; 90 }