csu1115 最短的名字 字典树模板

#include<iostream>
#include<cstring>
#include<cstdlib>
using namespace std;
const int M=26;  //一般是26个小写字母，可自行更改
typedef struct Trie{
Trie *next[M];
int v;  //标记  h
}Trie;
char s[1000001];
Trie *newTrie(){
Trie *cur=(Trie*)malloc(sizeof(Trie));
for(int i=0;i<M;i++)cur->next[i]=NULL;
cur->v=1;
return cur;
}
void insert(Trie *lin,char *word){
Trie *cur=lin;
for(int i=0;word[i]!='\0';i++){
if(cur->next[word[i]-'a']==NULL){
cur->next[word[i]-'a']=newTrie();
cur=cur->next[word[i]-'a'];
}
else{
cur=cur->next[word[i]-'a'];
cur->v++;
}
}
}
void Realse(Trie *lin){

for(int i=0;i<M;i++){
if(lin->next[i]!=NULL)Realse(lin->next[i]);
}
free(lin);
}
int search(Trie *lin){
int sum=0;
Trie *cur=lin;
for(int i=0;i<M;i++){
if(lin->next[i]!=NULL){
cur=lin->next[i];
sum+=cur->v;
if(cur->v>1)sum+=search(cur);
}
}
return sum;
}
int main(){
int t;
cin>>t;
while(t--){
Trie *lin=(Trie*)malloc(sizeof(Trie));
for(int i=0;i<M;i++)lin->next[i]=NULL;
lin->v=0;
int n;
cin>>n;
for(int i=0;i<n;i++){
cin>>s;
insert(lin,s) ;
}
cout<<search(lin)<<endl;
Realse(lin);
}
return 0;
}