HDU 2896 (AC自动机模板题)

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=2896

题目大意:多个模式串。多个匹配串。其中串的字符范围是(0~127)。问匹配串中含有哪几个模式串。

解题思路

AC自动机模板题。注意一下字符范围。

cnt记录这个模式串的个数改为这个模式串的index。

find的时候,把找到的index压入vector里面即可。

注意有多个匹配串,每次find之后会把last->cnt修改,原因是防止一个模式串出现了多次被压入vector,所以先备份一下,再还原回来。

#include "cstdio"
#include "cstring"
#include "string"
#include "iostream"
#include "queue"
#include "vector"
#include "algorithm"
using namespace std;
#define maxn 130
struct Trie
{
Trie *next[maxn],*fail;
int cnt;
}*root;
struct status
{
Trie *last;
int cnt;
status(Trie *last,int cnt):last(last),cnt(cnt) {}
};
Trie *newnode()
{
Trie *ret=new Trie;
memset(ret->next,,sizeof(ret->next));
ret->fail=;
ret->cnt=;
return ret;
}
void init() {root=newnode();}
void Insert(string str,int index)
{
Trie *pos=root;
for(int i=;i<str.size();i++)
{
int c=str[i];
if(!pos->next[c]) pos->next[c]=newnode();
pos=pos->next[c];
}
pos->cnt=index;
}
void getfail()
{
queue<Trie *> Q;
for(int c=;c<maxn;c++)
{
if(root->next[c])
{
root->next[c]->fail=root;
Q.push(root->next[c]);
}
else root->next[c]=root;
}
while(!Q.empty())
{
Trie *x=Q.front();Q.pop();
for(int c=;c<maxn;c++)
{
if(x->next[c])
{
x->next[c]->fail=x->fail->next[c];
Q.push(x->next[c]);
}
else x->next[c]=x->fail->next[c];
}
}
}
vector<int> find(string str)
{
Trie *pos=root,*last;
queue<status> Q;
vector<int> ans;
for(int i=;i<str.size();i++)
{
int c=str[i];last;
if(pos->next[c])
{
pos=pos->next[c];
last=pos;
while(last->cnt)
{
Q.push(status(last,last->cnt));
ans.push_back(last->cnt);
last->cnt=; //修改last->cnt
last=last->fail;
}
}
}
while(!Q.empty()) //恢复last->cnt
{
status x=Q.front();Q.pop();
x.last->cnt=x.cnt;
}
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
ios::sync_with_stdio(false);
int n,m;
string tt;
while(cin>>n)
{
init();
for(int i=;i<=n;i++)
{
cin>>tt;
Insert(tt,i);
}
getfail();
cin>>m;
int cnt=;
for(int i=;i<=m;i++)
{
cin>>tt;
vector<int> ans=find(tt);
sort(ans.begin(),ans.end());
if(!ans.size()) continue;
cnt++;
printf("web %d:",i);
for(int j=;j<ans.size();j++) printf(" %d",ans[j]);
printf("\n");
}
printf("total: %d\n",cnt);
}
}
2871595 neopenx HDU 2896 Accepted 29988 343 C++ 2565
9 min ago
上一篇:Codeforces Round #285 (Div. 2) A B C 模拟 stl 拓扑排序


下一篇:HDU 3065 (AC自动机模板题)