POJ 2001 Shortest Prefixes （字典树 TRIE）

题目链接：http://poj.org/problem?id=2001

【题目大意】

给你一些单词，让你求出他们最短的前缀，当然，这个前缀不能有歧义，例如给出单词 carton cart car

carton的 前缀 就不能是cart，因为cart的前缀是cart，同理cart的前缀也不能是car。

要找到每个单词独一无二且是最短的前缀，car的前缀不能是，”c“  “ca” ，因为他们在别的单词中也有出现，

如果找不到独一无二，那个这个单词本身就是他自己的前缀。

【样例输入】

carbohydrate
cart
carburetor
caramel
caribou
carbonic
cartilage
carbon
carriage
carton
car
carbonate

【样例输出】

carbohydrate carboh
cart cart
carburetor carbu
caramel cara
caribou cari
carbonic carboni
cartilage carti
carbon carbon
carriage carr
carton carto
car car
carbonate carbona

【思路】

将单词全部存入字典树

修改字典树的查找函数

void search(string s){
	int len =s.length();
	node *a=root;
	string ans=""; //记录prefixes
	for(int i=0;i<len;i++){
 		int k=s[i]-'a';
		 ans+=s[i];
		a=a->next[k];
		if(a->v==1){ //如果前缀唯一
			cout<<s<<" "<<ans<<endl;return ;
 		}
		if(a==NULL) return ;
	}
	cout<<s<<" "<<s<<endl; //否则输出本身
	// return a->v;
}

【源代码】

#include <iostream>
#include <cstdio>
#include <set>
#include <map>
#include <algorithm>
#include <cstring>
#include <string>
const int maxn = 1010;
using namespace std;
string Map[1010];
struct node{
    int v;
    node* next[26];
    node(){
        v=0;
        memset(next,0,sizeof(next));
    }
}*root;
void add(string word){
    int len=word.length();
    node*a=root,*b;
    for(int i=0;i<len;i++){
        int k=word[i]-'a';
        if(a->next[k]!=NULL){
            a=a->next[k];
            a->v++;
        }
        else{
            // cout<<"bug "<<endl;
            b=new node;
            b->v=1;
            a->next[k]=b;
            a=a->next[k];
        }
    }
}
void search(string s){
    int len =s.length();
    node *a=root;
    string ans="";
    for(int i=0;i<len;i++){
        int k=s[i]-'a';
        ans+=s[i];
        a=a->next[k];
        if(a->v==1){
            cout<<s<<" "<<ans<<endl;return ;
        }
        if(a==NULL) return ;
    }
    cout<<s<<" "<<s<<endl;
    // return a->v;
}
void clear(node *a){
    if(a==NULL)
        return ;
    else{
        for(int i=0;i<26;i++){
            clear(a->next[i]);
        }
    }
    delete(a);
}
int main(){
    string tmp;
    int cnt=0;
    root = new node;
    while(cin>>tmp){
        add(tmp);
        Map[cnt++]=tmp;
    }
    for(int i=0;i<cnt;i++){
        search(Map[i]);
    }
    clear(root);
    return 0;
}

另外本人写的暴力加排序代码也过了，分析后才知是后台数据水的缘故，还是要多多思考再敲代码。