下课前,园长提出了一个问题:“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) 对 1,000,000,007 取模的结果即可。
其中∏ (num[i] + 1)=1= (num[1] + 1) × (num[2] + 1) × ⋯ × (num[L] + 1)。
到头来还是到卡常题。。。
本来应该老老实实地用两次kmp的,偏偏写了一个SAM
当然也不难写,不过是在线在parent树上面做倍增而已(插入一个做一次倍增求答案)
我们对于新加入的节点np,只考虑它所有祖先中,sz[p]>0的节点(因为只有这些节点是串的前缀)
倍增到一个最深的节点p使得mx[p]*2<=mx[np]即可
因为只考虑sz[p]>0的祖先,所以求f[x][0]可以O(1),总体复杂度O(tNlogN)
让后开始无限卡常,用尽所学手段但是卡到现在依然没过,放弃。。。。。。
(讲道理,以前jz的机子都是跑的比我的电脑快的啊怎么今天比我这里还慢呢。。。1.2s怎么都卡不了啊)