一直觉得字典树是个高大上的东西,今天学了才发现也并没有那么难。。
附上讲解博客Orz:https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html
后来一直WA,应该是读数据的方法不对,,看了网上的博客发现gets是个好东西。。
附上AC代码:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAX=500010;
int trie[MAX][26];
int tot;
int sum[MAX];//被访问过的次数
void ins(char s[])
{
int len=strlen(s);
int root=0;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
if(!trie[root][id])
trie[root][id]=++tot;
root=trie[root][id];
sum[root]++;
}
}
int Search(char s[])
{
int len=strlen(s);
int root=0;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
if(!trie[root][id])
return 0;
root=trie[root][id];
}
return sum[root];
}
int main()
{
memset(trie,0,sizeof(trie));
memset(sum,0,sizeof(sum));
tot=0;
char s[13];
while(gets(s))
{
if(s[0]==NULL) //空行。gets读入的回车符会自动转换为NULL
break;
ins(s);
}
while(gets(s))
printf("%d\n",Search(s));
return 0;
}