题意:中文题,统计以某字符串作为前缀的字符串个数
刚学字典树,理解起来十分简单,就是维护一个多叉树,这里用的是链表版本,后面就用的是数组版本了,个人更喜欢数组版本,这里的链表版本就因为
莫名其妙的错误 C++能过而g++就会MLE 可能是两者管理内存的方式不一样吧
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=;
struct Trie{
Trie*next[maxn];
int flag;
Trie(){
flag=;
memset(next,,sizeof(next));
}
}*root;
void insert(char*str){
int len=strlen(str);
Trie*p=root,*q;
for(int i=;i<len;i++){
int id=str[i]-'a';
if(p->next[id]==NULL){
q=new Trie();
p->next[id]=q;
p=p->next[id];
}
else {
p=p->next[id];
(p->flag)++;
}
} }
int query(char*str){
int len=strlen(str);
Trie*p=root;
for(int i=;i<len;i++){
int id=str[i]-'a';
p=p->next[id];
if(p==NULL)return ;
}
return p->flag;
}
void Free(Trie*T){
if(T==NULL)return ;
for(int i=;i<maxn;i++){
if(T->next[i])Free(T->next[i]);
}
delete(T);
} int main(){
char temp[];
root=new Trie();
while(fgets(temp,,stdin)&&temp[]!='\n'){
temp[strlen(temp)-]='\0';
// cout<<temp<<endl;
insert(temp);
}
while(~scanf("%s",temp)){
printf("%d\n",query(temp));
}
Free(root);
return ;
}