【AC自动机模板】(转自木子日匀大神)

看了昀神的AC自动机,好棒啊~~~好棒啊。。。赶紧转过来了。。。。。


字典树上建失败指针什么的。

一个比较好的方法是稍微修改一下next的定义。

原来next[i][j]表示字典树中i节点的字符为j的儿子节点编号。

现在拓展一下。如果i节点没有字符为j的儿子,那么next[i][j]就是i节点沿着失败指针一直走到有字符为j的儿子的节点编号,如果没有,那么就指向根。

相当于是一个路径压缩的思想吧。这样一来,建立完自动机后,next指针就是很直接的转移关系了,不需要再特意的沿着失败指针跑啊跑的。

这种写法实现起来和用起来都很方便。

struct trie
{
    int next[1100][26],fail[1100],end[1100];
    int L,root;
    int newnode()
    {
        for (int i = 0;i < 26;i++)
            next[L][i] = -1;
        end[L++] = 0;
        return L-1;
    }
    void init()
    {
        L = 0;
        root = newnode();
    }
    void insert(char buf[],int val)
    {
        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] += val;
    }
    void build()
    {
        queue<int> Q;
        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
                {
                    fail[next[now][i]] = next[fail[now]][i];
                    Q.push(next[now][i]);
                }
        }
    }
};



你可能感兴趣的:(【AC自动机模板】(转自木子日匀大神))