ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

BZOJ1030 文本生成器

2020-10-21 22:32:00  阅读:156  来源: 互联网

标签:son 文本 int 生成器 next BZOJ1030 fail now root


pass

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 6010;
const int mod = 1e4 + 7;
int n, m;
int bin(int x, int y) {
    int ans = 1;
    for (; y; y >>= 1, x = (ll)x * x % mod)
        if (y & 1) ans = (ll)ans * x, ans %= mod;
    return ans;
}
struct Trie {
    int next[maxn][26], fail[maxn], end[maxn];
    int last[maxn];
    int dp[110][maxn];
    int root, L;
    int newnode() {
        for (int i = 0; i < 26; i++)
            next[L][i] = -1;
        fail[L] = last[L] = end[L] = 0;
        return L++;
    }
    void init() {
        L = 0;
        root = newnode();
    }
    void insert(char buf[]) {
        int len = strlen(buf);
        int now = root;
        for (int i = 0; i < len; i++) {
            if (next[now][buf[i] - 'A'] == -1)
                next[now][buf[i] - 'A'] = newnode();
            now = next[now][buf[i] - 'A'];
        }
        end[now]++;
    }
    void build() {
        queue<int> Q;
        fail[root] = root;
        for (int i = 0; i < 26; i++) {
            if (next[root][i] == -1)
                next[root][i] = root;
            else {
                fail[next[root][i]] = root;
                Q.push(next[root][i]);
            }
        }
        while (!Q.empty()) {
            int now = Q.front();
            Q.pop();
            end[now] += end[fail[now]];
            for (int i = 0; i < 26; i++) {
                if (next[now][i] == -1)
                    next[now][i] = next[fail[now]][i];
                else {
                    int son = next[now][i];
                    fail[son] = next[fail[now]][i];
                    last[son] = end[fail[son]] ? fail[son] : last[fail[son]];
                    Q.push(next[now][i]);
                }
            }
        }
    }
    void solve() {
        dp[0][0] = 1;
        for (int i = 1; i <= m; i++) {
            for (int j = 0; j < L; j++) {
                for (int k = 0; k < 26; k++) {
                    if (end[next[j][k]]) continue;
                    dp[i][next[j][k]] += dp[i - 1][j];
                    dp[i][next[j][k]] %= mod;
                }
            }
        }
        int ans = 0;
        for (int i = 0; i < L; i++) {
            ans += dp[m][i];
            ans %= mod;
        }
        printf("%d\n", ((bin(26, m) - ans) % mod + mod) % mod);
    }
} ac;
char s[maxn];
int main() {
    ac.init();
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i++) {
        scanf("%s", s);
        ac.insert(s);
    }
    ac.build();
    ac.solve();
    return 0;
}

标签:son,文本,int,生成器,next,BZOJ1030,fail,now,root
来源: https://www.cnblogs.com/Zeronera/p/13855293.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有