hiho刷题日记——第四天Trie图

用和kmp相似的思路，将trie树变为trie图。以用于在长篇文章中查找是否存在字典中的单词。

在此强调本文中的‘根’和一般的树中的‘根’不一样！

1.建立trie树

①建立一个结点结构体和初始化函数

typedef struct node{

bool isEnd;//结点是否为单词的结尾

struct node* trie;//结点的根（这里的根和一般的树的根不一样，但是怎么解释呢。不知道啊。）

struct node* next[26];

}TrieNode;

void initNode(TrieNode *p)

{

p->isEnd = 0;

for(int i=0;i<26;i++)

p->next[i]=NULL;

}

②建立一个head结点并初始化。

③建立一个headTrie结点作为head的根，并将其所有的next指向head。

④对于输入单词的处理：

输入一个字符串并扫描，只要还没结束：
并判断是否存在对应字符的子结点。
若存在，转到子结点；否者建一个子结点并转到该结点。
当单词扫描结束时，将最后一个结点的isEnd属性标为true。
void makeTrie(char *w)

{

int i=0;

TrieNode* p=&head;

while(w[i])

{

if(p->next[w[i]-97])

{

p=p->next[w[i]-97];

}

else

{

TrieNode *q=(TrieNode *)malloc(sizeof(TrieNode));

initNode(q);

p->next[w[i]-97]=q;

p=q;

}

i++;

}

p->isEnd=1;

}

2.建立trie图

用广度遍历。将&head加入遍历队列，并将head.trie赋值为&headTrie作为head的根。

只要遍历队列里还有元素，则做一下操作：

①先取出元素。

②判断是否存在对应字符的子结点。

若有，则将其子结点加入遍历队列，并将其子结点的根结点设为其根结点的对应字符的子结点。。。。。文字好绕，上代码p->next[i]->trie=p->trie->next[i];

若无，则将其对应的子结点设为其根结点的对应字符的子结点。p->next[i]=p->trie->next[i];

这里就是用的kmp的思想。

void makeTrieMap()

{

int front=0,rear=0;

queue[rear++]=&head;

head.trie=&headTrie;

while(front<rear)

{

TrieNode* p=queue[front++];

for(int i=0;i<26;i++)

{

if(p->next[i]!=NULL)

{

queue[rear++]=p->next[i];

p->next[i]->trie=p->trie->next[i];

}

else

{

p->next[i]=p->trie->next[i];

}

}

}

}

3.使用trie图

扫描文章。

从head开始对比，一直照着next来跳转。直达有一个结点的isEnd为true。说明找到字典中的某个单词了。返回true；

否则直到扫描结束都没找到单词，返回false。

bool find(char *s)

{

int i=0;

TrieNode* p=&head;

while(s[i])

{

p=p->next[s[i]-97];

if(p->isEnd) return true;

i++;

}

return false;

}

题目全代码：

#include<cstdio>

#include<cstdlib>

using namespace std;

typedef struct node{

bool isEnd;

struct node* trie;

struct node* next[26];

}TrieNode;

TrieNode head;

TrieNode headTrie;

TrieNode* queue[1000000+1];

void initNode(TrieNode *p)

{

p->isEnd = 0;

for(int i=0;i<26;i++)

p->next[i]=NULL;

}

void makeTrie(char *w)

{

int i=0;

TrieNode* p=&head;

while(w[i])

{

if(p->next[w[i]-97])

{

p=p->next[w[i]-97];

}

else

{

TrieNode *q=(TrieNode *)malloc(sizeof(TrieNode));

initNode(q);

p->next[w[i]-97]=q;

p=q;

}

i++;

}

p->isEnd=1;

}

void makeTrieMap()

{

int front=0,rear=0;

queue[rear++]=&head;

head.trie=&headTrie;

while(front<rear)

{

TrieNode* p=queue[front++];

for(int i=0;i<26;i++)

{

if(p->next[i]!=NULL)

{

queue[rear++]=p->next[i];

p->next[i]->trie=p->trie->next[i];

}

else

{

p->next[i]=p->trie->next[i];

}

}

}

}

bool find(char *s)

{

int i=0;

TrieNode* p=&head;

while(s[i])

{

p=p->next[s[i]-97];

if(p->isEnd) return true;

i++;

}

return false;

}

int main()

{

int N;

char word[1000000+1];

char str[1000000+1];

initNode(&head);

for(int i=0;i<26;i++)

headTrie.next[i]=&head;

scanf("%d",&N);

while(N--)

{

scanf("%s",word);

makeTrie(word);

}

makeTrieMap();

scanf("%s",str);

if(find(str)) printf("YES");

else printf("NO");

return 0;

}