hdu2027 trie树 字典树模板

#include <iostream>
#include <cstdio>
#include <cstring>
#include <sstream>
using namespace std;

#define Max 26

const int maxn=;

typedef struct TrieNode{
    int nCount;//根据需要改变
    struct TrieNode *next[Max];
}TrieNode;

TrieNode *root=NULL;

int allcop=,ans,flag;

TrieNode memory[maxn];

TrieNode *CreatTrieNode(){
     TrieNode *temp=&memory[allcop++];
     temp->nCount=;
     for(int i=;i<Max;i++) temp->next[i]=NULL;
     return temp;
}

void InsertTrie(TrieNode **pRoot,char *str){
      TrieNode *temp=*pRoot;
      int i=,k;
      while(str[i]){
        k=str[i]-'a';
        if(temp->next[k]){
                //temp->next[k]->nCount++;
        } else temp->next[k]=CreatTrieNode();
        temp=temp->next[k];
        i++;
      }
      temp->nCount++;
}

int SearchTrie(TrieNode *root,char *str){
    if(root==NULL) return ;
    TrieNode *temp=root;
      int i=,k;
      while(str[i]){
        k=str[i]-'a';
        if(temp->next[k]){
                temp=temp->next[k];
        } else return ;
        i++;
      }
      return temp->nCount;
}

void Traverse(TrieNode *root){
     for(int i=;i<Max;i++){
        if(root->next[i]){
                if(root->next[i]->nCount>) ans++;
                Traverse(root->next[i]);
        }
     }
}

char str[];

void init(){
  ans=;
  flag=;
  allcop=;
  memset(memory,,sizeof(memory));
  root=CreatTrieNode();
}

int main()
{
    while(){
        init();
        string line;
        getline(cin,line);
        stringstream ss(line);
        while(ss>>str) {
        if(str[]=='#') {
                flag=;
                break;
        }
        InsertTrie(&root,str);
        }
        if(flag) break;
        Traverse(root);
        printf("%d\n",ans);
    }
    return ;
}

输入方式值得学习