BZOJ3670 [Noi2014]动物园

哇，你造吗。。。蒟蒻当年NOI这道题。。。可是拿了0分哦~（废话这么多，你弱怪我啊！）

我们先kmp一次，记录下next数组及cnt数组，其中cnt表示他前面可以匹配的模式串个数。

然后在类似kmp的做一次，记录下Next数组，Next == next + 长度限制。

于是。。。ans = Π (cnt[Next[i]] + 1) (1 ≤ i ≤ len)

/**************************************************************
    Problem: 3670
    User: rausen
    Language: C++
    Result: Accepted
    Time:664 ms
    Memory:13500 kb
****************************************************************/
 
#include <cstdio>
#include <cstring>
 
using namespace std;
typedef long long ll;
const int Len = 1000005;
const ll mod = 1000000007;
 
char st[Len];
int len, next[Len], Next[Len], cnt[Len];
ll ans;
 
inline void work_next() {
    int i, t;
    for (i = 1; i <= len; ++i) {
        t = next[i - 1];
        while (t != -1 && st[i] != st[t + 1])
            t = next[t];
        next[i] = t + 1, cnt[i] = cnt[t + 1] + 1;
    }
}
 
inline void work_Next() {
    int i, t;
    for (i = 1; i <= len; ++i) {
        t = Next[i - 1];
        if (t * 2 + 2 > i) t = next[t];
        while (t != -1 && st[i] != st[t + 1])
            t = next[t];
        Next[i] = t + 1;
    }
}
 
int main() {
    int T, i;
    scanf("%d
", &T);
    while (T--) {
        gets(st + 1), len = strlen(st + 1);
        next[0] = Next[0] = -1;
        work_next();
        work_Next();
        for (i = ans = 1; i <= len; ++i)
            (ans *= (cnt[Next[i]] + 1)) %= mod;
        printf("%lld
", ans);
    }
}