最短前綴（字典樹）

剛學了字典樹，找了一道簡單的模板題做了一下，下面每行前面部分的字符串是輸入的，後面是其最小

前綴，下面是要求的輸出格式：

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

代碼及註釋如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>

using namespace std;

//樹的結構
struct Trie
{
    int cnt;  //每個字母是 cnt 個單詞的前綴
    Trie *next[26]; //一個指針數組，指向他下一個字母；
    
    Trie()
    {
        cnt = 0;
        for(int i = 0 ; i < 26 ; i ++)
            next[i] = NULL;
    }
};
Trie *root ;      //樹根結點
char a[1005][25]; //記錄所有字符串，以便輸出

//每次傳入一個字符串建樹。
void createTrie(char *str)
{
    int len = strlen(str);
    Trie *p = root;
    Trie *q;                 
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'a';
        if(p->next[id] == NULL)
        {
            q = new Trie;
            q->cnt = 1;
            p->next[id] = q;
        }
        else
        {
            p->next[id]->cnt++;
        }
        p = p->next[id];
    }
}

//查找某個字符串對應的最小前綴，並返回每個字符串對應最小前綴的長度
int findTrie(char *str)
{
    int i;
    int len = strlen(str);
    Trie *p = root;
    int cnt = 0;
    for(i=0; i<len; ++i)
    {
        int id = str[i]-'a';
        p = p->next[id];
        if(p -> cnt > 1 )
            cnt++;
        else
            break;
    }
    if(i == len)     
        return cnt;
    else
        return ++cnt;  //需要把第一次出現cnt爲1時的節點也加入最小前綴
}
int main()
{
    int n=0;
    root = new Trie;
    while(scanf("%s", a[n])!=EOF)
    {
        getchar();
        createTrie(a[n]);
        n++;
    }
    for(int i=0; i<n; ++i)
    {
        cout << a[i] << " ";
        int len = findTrie(a[i]);
        for(int j=0; j<len; ++j)
            printf("%c", a[i][j]);
        cout << endl;
    }


    return 0;
}