HDU_2222

? ? 今天開始學AC自動機了，這個就是我AC自動機的處女作了。這個題有個小trick就是單詞列表中可能有重復的單詞，但這些重復的單詞應看做不同的，因此建字典樹時做標記的時候，把原來的賦值為1的操作變為自加1的操作即可。

? ? 最后匹配的時候有一個小優化，就是比如當前正在看有多少個以j結點的字符為結尾的單詞，那么我們還需要不停地沿預處理的標記向上找，當我們找過之后就可以把當前節點的標記做成-1，以后再遇到-1的時候，就直接break，無需再沿著標記向上找了，因為上面的結點一定在之前就找過了。這個優化大概能節省400ms的時間。

#include<stdio.h>
#include<string.h>
#define MAXD 500010
#define MAXT 1000010
char b[60], txt[MAXT];
int N, next[MAXD][26], flag[MAXD], P[MAXD], q[MAXD], e, cnt;
void insert(int cur, int k)
{
++ e;
flag[e] = 0;
memset(next[e], 0, sizeof(next[e]));
next[cur][k] = e;
}
void init()
{
int i, j, k, cur;
scanf("%d", &N);
e = 0;
memset(next[e], 0, sizeof(next[e]));
for(i = 0; i < N; i ++)
{
scanf("%s", b);
cur = 0;
for(j = 0; b[j]; j ++)
{
k = b[j] - 'a';
if(!next[cur][k])
insert(cur, k);
cur = next[cur][k];
}
++ flag[cur];
}
}
void prepare()
{
int i, j, k, u, x, front, rear;
front = rear = 0;
P[0] = q[rear ++] = 0;
while(front < rear)
{
u = q[front ++];
for(i = 0; i < 26; i ++)
if(next[u][i])
{
x = next[u][i];
if(u == 0)
P[x] = 0;
else
{
for(j = P[u]; j != 0; j = P[j])
if(next[j][i])
{
P[x] = next[j][i];
break;
}
if(j == 0)
P[x] = next[0][i];
}
q[rear ++] = next[u][i];
}
}
}
void solve()
{
int i, j, k, t;
prepare();
scanf("%s", txt);
j = cnt = 0;
for(i = 0; txt[i]; i ++)
{
k = txt[i] - 'a';
while(j > 0 && !next[j][k])
j = P[j];
j = next[j][k];
for(t = j; t > 0; t = P[t])
{
if(flag[t] >= 0)
{
cnt += flag[t];
flag[t] = -1;
}
else
break;
}
}
printf("%d\n", cnt);
}
int main()
{
int t;
scanf("%d", &t);
while(t --)
{
init();
solve();
}
return 0;
}

轉載于:https://www.cnblogs.com/staginner/archive/2012/01/13/2321760.html

總結

以上是生活随笔為你收集整理的HDU 2222 Keywords Search的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。