HDU1671字典树

这题纠结了好久，之前没有考虑到flag标记，如果有如下的数据，之前的想法就错了:

1

2

00000

000

 

直接建字典树，注意一下几点：

1.如果当前的结点是标记了danger的，那么就直接返回false，表示有前缀。

2.如果遍历到最后依然没有发现danger标记，要考虑这个结点有没有后继，如果有，说明他是某些结点的前缀，这个用flag标记。

 

自己在代码里加了一些注释，以便理解。

我的代码：

 

#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[10]; bool danger,flag; //if node has next pointer set the flag true }node[MAX],*root; int K; Node* New(){ //get a new pointer which has not be allocate. Node* ret=&node[K++]; for(int i=0;i<10;i++){ ret->ne[i]=NULL; } ret->danger=false; ret->flag=false; return ret; } void init(){ K=0; root=New(); } bool insert(char* s){ Node* ptr=root; char* p=s; int id; //go along the tree and find whether //s has a dangerous prefix. while(*p){ id=*(p++)-'0'; if(ptr->danger){ return false; } if(ptr->ne[id]==NULL){ ptr->ne[id]=New(); } ptr->flag=true; //set the flag true because this node has next pointer. ptr=ptr->ne[id]; } ptr->danger=true; //notice: if the next pointer is not null //we say s is a prefix of some words. if(ptr->flag){ return false; } return true; } int main(){ int t; int n; bool done; char s[1100]; scanf("%d",&t); while(t--){ init(); scanf("%d",&n); done=true; for(int i=0;i<n;i++){ scanf("%s",s); if(done){ done=insert(s); } } if(done){ puts("YES"); }else{ puts("NO"); } } return 0; }