Trie树
Trie树是一个时间换空间的字符串结构。它是一个二十六叉树。它是这样存字符串的:
比如对于一个字符串“abcd”,它先看看根节点相连的点有没有a,那就新开一个点;继续找,看看a相连的点有没有b,重复上述操作……
然后为了避免一个字符串是另外一个字符串的前缀的问题,我们再在每一个结点上面新开一个域,就是endflag域,代表有一个字符串在这里结束。
HLoj611 Trie树
code:
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
int num=0;
char c=' ';
bool flag=true;
for(;c>'9'||c<'0';c=getchar())
if(c=='-')
flag=false;
for(;c>='0'&&c<='9';num=num*10+c-48,c=getchar());
return flag ? num : -num;
}
string s;
int len;
struct node
{
char ch;
bool endflag;
int link[26];
}tree[600100];
void invise(int k,int node)
{
int letter=s[k]-'A';
if(tree[node].link[letter]==0)
{
tree[node].link[letter]=++len;
tree[len].ch=s[k];
tree[len].endflag=false;
}
int nextnode=tree[node].link[letter];
if(k==s.size()-1)
{
tree[len].endflag=true;
return;
}
invise(k+1,nextnode);
}
bool find(int k,int node)
{
int letter=s[k]-'A';
if(tree[node].link[letter]==0)
return false;
int nextnode=tree[node].link[letter];
if(k==s.size()-1)
{
if(tree[node].endflag)return true;
else return false;
}
return find(k+1,nextnode);
}
int main()
{
while(getline(cin,s))
{
invise(0,0);
}
printf("%d\n",len+1);
return 0;
}