hiho刷题日记——第二天Trie树

建立Trie树来统计以某字符串为前缀的单词在字典中的个数。

思路：

1.Trie树的构建

①先写一个结点的结构体和结点的初始化函数

typedef struct node{
int count;//因为题目是统计以某字符串为前缀的单词在字典中的个数，所以在这里做统计了。
struct node* next[26];
}TrieNode; 

void initNode(TrieNode *p)
{
p->count = 0;
for(int i=0;i<26;i++)
p->next[i]=NULL;
}

②建一个根结点head并初始化。//现在想起写root才对啊- -。当时写的head。

③对于输入单词的处理：

输入一个字符串并扫描，只要还没结束：

      当前结点的count加一，并判断是否存在对应字符的子结点。

      若存在，转到子结点；否者建一个子结点并转到该结点。

void makeTrie(char *w)
{
int i=0;
TrieNode* p=&head;

while(w[i])
{
p->count++; 
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->count++;
}

2.Trie树的运用

对于要查找的字符串进行扫描

从head开始对比，如果存在对应字符的子结点，则转到该子结点。

否者就没有以该字符串为前缀的单词，返回0。

当字符串扫描完依然没有返回0，则返回当前结点的count，即以改字符串为前缀的单词数量。

int find(char *w)
{
int i=0;
TrieNode* p=&head;
while(w[i])
{
if(p->next[w[i]-97])
{
p=p->next[w[i]-97];
i++;
}
else
{
return 0;
}
}
return p->count;
} 

全代码：

#include<cstdio>
#include<cstdlib>
using namespace std;


typedef struct node{
int count;
struct node* next[26];
}TrieNode; 


TrieNode head;


void initNode(TrieNode *p)
{
p->count = 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])
{
p->count++; 
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->count++;
}


int find(char *w)
{
int i=0;
TrieNode* p=&head;
while(w[i])
{
if(p->next[w[i]-97])
{
p=p->next[w[i]-97];
i++;
}
else
{
return 0;
}
}
return p->count;
} 


int main()
{
int N;
char word[12];


initNode(&head);

scanf("%d",&N);
while(N--)
{
scanf("%s",word);
makeTrie(word);
}

scanf("%d",&N);
while(N--)
{
scanf("%s",word);
printf("%d\n",find(word));
}
return 0;
}