您的位置:首页 > 其它

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

2014-06-30 19:05 204 查看
看了昀神的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]);
}
}
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: