【hihocoder】#1014 : Trie树

题目链接

题目思路

代码如下

#include <iostream>

using namespace std;

const int maxn = 10+5;
struct TrieNode
{
int num;
TrieNode* next[26];
TrieNode(){
num=0;
for(int i=0;i<26;i++)
{
next[i]=NULL;
}
}

};
struct TrieNode root;
void Insert(char ch[])
{
struct TrieNode *p = &root;

for(int i=0;ch[i];i++)
{
int idx = ch[i]-'a';
if(p->next[idx]==NULL)
p->next[idx]=new struct TrieNode;
p=p->next[idx];
p->num++;
}

}
int Select(char ch[])
{
struct TrieNode *p = &root;
for(int i=0;ch[i];i++)
{
int idx = ch[i]-'a';
if(p->next[idx]==NULL)
return 0;
p=p->next[idx];
}
return p->num;
}
int main()
{
int n,m;
cin>>n;
char ch[maxn];
for(int i=0;i<n;i++)
{
cin>>ch;
Insert(ch);
}
cin>>m;
for(int i=0;i<m;i++)
{
cin>>ch;
cout<<Select(ch)<<endl;
}

return 0;
}