字符串唯一前缀问题；一道Trie树笔试题；

一个文件里面有如下字符串
cartefdxh
cart
carlkijfwe
chdfwef
cafkekfld
............

要从文件中找出唯一能代表该字符串的前缀，然后如下输出
cartefdxh carte
cart cart
carlkijfwe carl
chdfwef ch
cafkekfld caf

以空格分隔.......

 

 

#include <iostream>#include <vector>#include <string>using namespace std;struct Trie{int flag;struct Trie* next[26];void init(){memset(next,0,sizeof(next));flag=0;}}trie[10000];int k=1;struct Trie *root=&trie[0];void insert(string &str){struct Trie *p=root;int i=0;int j;while(str[i]!='/0'){j=str[i]-'a';if(p->next[j]==0){trie[k].init();p->next[j]=&trie[k++];}p=p->next[j];++p->flag;++i;}}int find(string &str){struct Trie *p=root;int i=0,j;while(str[i]!='/0'){j=str[i]-'a';p=p->next[j];if(p->flag==1){return i+1;}++i;}return i;}int main(){vector<string> v;string str;root->init();int n;cin>>n;for(int i=0;i<n;++i){cin>>str;v.push_back(str);insert(str);}for(int j=0;j<n;++j){cout<<v[j]<<" ";int k=find(v[j]);cout<<v[j].substr(0,k)<<endl;}return 0;}

 

 

Trie的灵活使用,给每个结点一个标记flag,如果flag==1，说明只有一个单词以这个前缀为前缀,就找到了.

如果找到单词末尾,都未找到flag==1的，说明没有唯一前缀,打印出自身表示无奈!

 

效率真是优秀,内存量很大,思想很好,学习了.