poj 2001

http://poj.org/problem?id=2001

大意: 求一字符串的最短前缀

解题思路：trie树的简单应用。。直接模版即可

#include <iostream>
#include<cstring>
using namespace std;
struct node{
    int cnt;
    struct node *next[26];
    node(){//结构体中可以带函数，进行初始化
        cnt =0;
        memset(next,0,sizeof(next));
    }
};
char str[1003][30];
node *root = NULL;//初始化，，根节点不用。。

void maketrie(char *s){
    node *p = root;
    node *temp = NULL;
    for(int i=0;i<strlen(s);i++){
        if(p->next[s[i]-'a']==NULL){//查看节点中是否有此节点。。没有的话加上，有的话。。cnt++;
            temp = new node;
            p->next[s[i]-'a']=temp;
        }
        p = p->next[s[i]-'a'];
        p->cnt++;
    }
}

void search(char *s){//输出。。如果其cnt==1 证明他是该字符串的最短前缀，没有于别的字符串公用该节点。
    node *p =root;
    for(int i=0;i<strlen(s);i++){
        p = p->next[s[i]-'a'];
        cout<<s[i];
        if(p->cnt==1)
            break;
    }
}

int main()
{
    int num = 0;
    root = new node;
    while(cin>>str[num]){
        maketrie(str[num]);
        num++;
    }

    for(int i=0;i<num;i++){
        cout<<str[i]<<" ";
        search(str[i]);
        cout<<endl;
    }
    return 0;
}