POJ2001字典树

这题在cnt变量这里卡了好久，一直WA着，才发觉不对。、

 

和常规的题思路一致，查询的时候找到第一个cnt为1的结点就是解，注意insert函数cnt的更新必须更新next指针，root单独更新，如果要到此结点了更新cnt，那么最后的一个结点要更新cnt，如果不更新会发生错误。

 

我的代码，加上了注释：

 

#include <vector> #include <list> #include <map> #include <set> #include <deque> #include <queue> #include <algorithm> #include <stack> #include <bitset> #include <functional> #include <numeric> #include <utility> #include <sstream> #include <iostream> #include <iomanip> #include <cstdlib> #include <cstdio> #include <cstring> #include <cctype> #include <string> #include <ctime> #include <cmath> using namespace std; const int MAX=1000000; struct Node{ Node* ne[26]; int cnt; }node[MAX],*root; int K; Node* New(){ Node* ret=&node[K++]; ret->cnt=0; for(int i=0;i<26;i++){ ret->ne[i]=NULL; } return ret; } void init(){ K=0; root=New(); } void insert(char* s){ Node* ptr=root; char* p=s; int id; ptr->cnt++; //here we update the cnt at first when we reach it. while(*p){ id=*(p++)-'a'; if(ptr->ne[id]==NULL){ ptr->ne[id]=New(); } ptr=ptr->ne[id]; ptr->cnt++; } } char* find(char* s){ char* p=s; static char ret[30]; int top=0; Node* ptr=root; int id; while(*p){ if(ptr->cnt==1){ ret[top]=0; return ret; } ret[top++]=*p; id=*(p++)-'a'; ptr=ptr->ne[id]; } return s; } char s[1100][30]; int main(){ int n=0; init(); while(gets(s[n])){ insert(s[n++]); } for(int i=0;i<n;i++){ printf("%s %s/n",s[i],find(s[i])); } return 0; }