题目:
http://hihocoder.com/problemset/problem/1036
题意:
给定多个模式串,问原串中有没有包含至少一个模式串
思路:
trie图应用,题目上有详细讲解,hiho一下第四周的题目,可以看大佬的代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include <vector>
#include <queue>
using namespace std;
const int N = 1000000 + 10, INF = 0x3f3f3f3f;
struct node
{
node *next[26];
node *suff; //指向后缀节点
bool flag;
}trie[N], *root;
int tot;
char ori[N];
node* node_init()
{
trie[tot].flag = false;
trie[tot].suff = NULL;
memset(trie[tot].next, 0, sizeof trie[tot].next);
return trie + tot++;
}
void trie_insert(char *s)
{
node *p = root;
for(int i = 0; s[i]; i++)
{
int j = s[i] - 'a';
if(p->next[j] == NULL) p->next[j] = node_init();
p = p->next[j];
}
p->flag = true;
}
void trie_graph()
{//trie[0]为虚拟节点,root为trie[1],trie[0]的所有边均指向root,方便以后操作
for(int i = 0; i < 26; i++) trie[0].next[i] = root;
root->suff = trie + 0;
trie[0].suff = NULL;
queue<node*> que;
que.push(root);
while(! que.empty())
{
node *p = que.front(); que.pop();
for(int i = 0; i < 26; i++)
if(p->next[i])
{//查看父亲节点的后缀节点是否存在编号为i的边,若没有,就一直往上找,直到根节点
node *ptr = p->suff;
while(ptr && ! ptr->next[i]) ptr = ptr->suff;
//要么找到了,要么循环到了虚拟节点后,后缀节点被置为root
p->next[i]->suff = ptr->next[i];
if(ptr->next[i]->flag) p->next[i]->flag = true;
que.push(p->next[i]);
}
}
}
bool trie_query(char *s)
{
node *p = root;
for(int i = 0; s[i]; i++)
{
int j = s[i] - 'a';
while(true)
{
if(p->next[j] != NULL)
{
p = p->next[j];
if(p->flag == true) return true;
break;
}
else p = p->suff;
}
}
return false;
}
int main()
{
int n;
scanf("%d", &n);
tot = 2;//tot+0是虚拟节点,tot+1是根节点,故从2开始
root = trie + 1;
memset(trie[root].next, 0, sizeof trie[root].next);
for(int i = 1; i <= n; i++)
{
scanf("%s", ori);
trie_insert(ori);
}
trie_graph();//建trie图
scanf("%s", ori);
puts(trie_query(ori) ? "YES" : "NO");
return 0;
}