POJ 3630 Phone List

Trie的应用，判断一个串是否为另一个串的前缀

需要注意的是前缀可能先出现，也可能后出现

还有就是需要静态建树，否则会TLE

 

插入和查找可以一起实现，另外MAX从500000变到80000的时候时间从500ms变到157ms了，^_^

 

代码：

#include<iostream> #include<cstdio> #include<memory.h> #include<string> using namespace std; #define base const int MAX=100005; struct trie { bool end; int next[11]; }trie_node[MAX]; int index,cnt; bool insert(char str[]) { int i=0,del,p=0; bool flag=true; while(str[i]!='/0') { del=str[i]-'0'; if(trie_node[p].next[del]==NULL) { trie_node[p].next[del]=++index; if(trie_node[p].end==true)//别的字符串时当前串的前缀 flag=false; } trie_node[p].next[10]=1;//表示访问过 p=trie_node[p].next[del]; i++; } trie_node[p].end=true; if(trie_node[p].next[10])//当前串是别的串的前缀 flag=false; return flag; } /*bool search(char word[]) { int i=0,t,in=0;//root while(word[i]!='/0') { t=word[i]-'0'; if(trie[in].next[t]==0) { memset(trie[nodeindex].next,0,sizeof(trie[nodeindex].next)); trie[nodeindex].key=0; trie[in].next[t]=nodeindex++; } in=trie[in].next[t]; if(word[i+1]=='/0'&&trie[in].key>0)//是别的串的前缀 return false; if(word[i+1]!='/0'&&trie[in].key==2)//别的串是它的前缀 return false; trie[in].key=1;//已经访问过 i++; } trie[in].key=2;//串结束 return true; }*/ int main() { int i,T,n; char s[15]; scanf("%d",&T); while(T--) { cnt=index=0; memset(trie_node,0,sizeof(trie_node)); trie_node[0].end=false; scanf("%d",&n); bool possible=true; while(n--) { scanf("%s",&s); if(possible) { possible=insert(s); } /*bool y=insert(s); cout<<y<<endl; if(!y) possible=false; */ } if(possible) puts("YES"); else puts("NO"); } return 0; }