POJ2001 字典树（模版题）

                                                                   ——头一次写博客，心里很没有底

     嗯。。前几天跟师父学了字典树等一系列字符串类问题，说到底也就明白个大概，上手着实费了个劲。

好在有师父“喂”题，不然真的是一头雾水。

     好吧，回到这道题上，题意要求是构建一棵Trie树，然后找到每个能确定该字符串的最短前缀。就如标题所说，红果果的模版题。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
struct Trie
{
    int cnt;
    Trie *p[26];
}*root;
char a[1010][30];
void MakeTree(char *s)//建树模版
{
    Trie *r=root;
    Trie *tmp;
    for(int i=0;i<strlen(s);i++)
    {
        if(r->p[s[i]-'a']==NULL)
        {
            tmp=new Trie;
            tmp->cnt=0;   //初始cnt==0
            for(int j=0;j<26;j++)
                tmp->p[j]=NULL;
            r->p[s[i]-'a']=tmp;
        }
        r->p[s[i]-'a']->cnt++;
        r=r->p[s[i]-'a'];
        /*r=r->p[s[i]-'a'];
        r->cnt++;*///两者意义一样，写法均可
    }
}
void FindTrie(char *s)//查找树模版
{
    int len=strlen(s);
    Trie *r=root;
    int i;
    for(i=0;i<len;i++)
    {
        r=r->p[s[i]-'a'];
        printf("%c",s[i]);
        if(r->cnt==1)//直到找到不存在以此为前缀的串
            break;
    }
}
int main()
{
    root=new Trie;
    for(int i=0;i<26;i++)
        root->p[i]=NULL;
        //(*root).p[i]=NULL;结构体指针变量->成员名，也是，两者均可
    int n;
    for(n=0;;n++)
    {
        if(cin>>a[n])
            MakeTree(a[n]);
        else
            break;
    }
    for(int i=0;i<n;i++)
    {
        printf("%s ",a[i]);
        FindTrie(a[i]);
        printf("\n");
    }
    return 0;
}

开开心心的A了之后，又看了眼师父当年写的代码，表示又看不懂了= =

也罢。。。一点点来吧，先在下面存一个，以供日后再比较着看。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
struct trie{
    int ch[100100][30];
    int num[100100];
    int sz;
    void reset(){
        memset(ch[0], 0, sizeof(ch[0]));
        memset(num, 0, sizeof(num));
        sz = 1;
    }
    int idx(char c){
        return c - 'a';
    }
    void insert(string s){
        int u = 0, n = s.size();
        for (int i = 0; i < n; i++){
            int c = idx(s[i]);
            if (!ch[u][c]){
                ch[u][c] = sz;
                memset(ch[sz], 0, sizeof(sz));
                num[sz] = 1; sz += 1;
            }
            else num[ch[u][c]] += 1;
            u = ch[u][c];
        }
    }
    void print(string s){
        int u = 0, n = s.size();
        for (int i = 0; i < n; i++){
            int c = idx(s[i]);
            putchar(s[i]);
            if (num[ch[u][c]] == 1) return;
            u = ch[u][c];
        }
    }
}t;
string s[2010];
int main(){
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    int n = 0;
    t.reset();
    while (cin >> s[n]) t.insert(s[n++]);
    for (int i = 0; i < n; i++){
        cout << s[i] << " ";
        t.print(s[i]);
        printf("\n");
    }
    return 0;
}

转载于:https://www.cnblogs.com/gaoxiang36999/p/4451532.html