zju2346 Shortest Prefixes(字典树 trie)

总算自己做出一道字典树，内牛满面~
题意是给你一串单词，让你找出最短的前缀，使得该前缀能唯一识别该单词
我的方法是建立字典树，并给每个节点的使用次数赋值。
从单词的第一个字母开始找，若找到某个节点使用次数为1，则输出到该节点就是最短前缀。
若没有一个是为1的节点，就全部输出了，题意已说明。

 

#include<iostream> #include<cstdlib> #include<cstdio> #include<cmath> #include<algorithm> #include<vector> #include<string> #include<sstream> #include<set> #include<map> #include<queue> #include<stack> using namespace std; const int number=26; int Case=1; class Trie { public: int num; Trie *child[number]; Trie() { num=0; for(int i=0;i<number;i++) { child[i]=0; } } }; void Insert(Trie *head,char *str) { Trie *p=head; int len=strlen(str); for(int i=0;i<len;i++) { int x=str[i]-'a'; if(p->child[x]==0) { p->child[x]=new Trie(); } p=p->child[x]; p->num++; } } void run() { Trie *head=new Trie(); char str[21]; char word[10001][21]; int k=0; while(gets(str)) { if(strcmp(str,"")==0) break; strcpy(word[k++],str); Insert(head,str); } for(int i=0;i<k;i++) { Trie *p=head; printf("%s ",word[i]); int len=strlen(word[i]); for(int j=0;j<len;j++) { int x=word[i][j]-'a'; p=p->child[x]; if(p->num==1) { printf("%c",word[i][j]); break; } else { printf("%c",word[i][j]); } } printf("/n"); } } int main() { int t; scanf("%d",&t); cin.get(); cin.get(); while(t--) { if(Case>1) printf("/n"); run(); Case++; } return 0; }