字典树 06:最短前缀

06:最短前缀

一个字符串的前缀是从该字符串的第一个字符起始的一个子串。例如 "carbon"的字串是: "c", "ca", "car", "carb", "carbo", 和 "carbon"。注意到这里我们不认为空串是字串, 但是每个非空串是它自身的字串. 我们现在希望能用前缀来缩略的表示单词。例如, "carbohydrate" 通常用"carb"来缩略表示. 现在给你一组单词, 要求你找到唯一标识每个单词的最短前缀
在下面的例子中，"carbohydrate" 能被缩略成"carboh", 但是不能被缩略成"carbo" (或其余更短的前缀) 因为已经有一个单词用"carbo"开始
一个精确匹配会覆盖一个前缀匹配，例如，前缀"car"精确匹配单词"car". 因此 "car" 是 "car"的缩略语是没有二义性的 , “car”不会被当成"carriage"或者任何在列表中以"car"开始的单词.

输入

输入包括至少2行，至多1000行. 每行包括一个以小写字母组成的单词，单词长度至少是1，至多是20.

输出

输出的行数与输入的行数相同。每行输出由相应行输入的单词开始，后面跟着一个空格接下来是相应单词的没有二义性的最短前缀标识符。

样例输入

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

代码实现

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <map>
#include <vector>
#include <set>
#include <cmath>
#include <algorithm>

using namespace std;

struct Tree                            //字典树，每个节点下有26个分支，分别代表abcde...；
{
    int cnt;
    Tree *next[26];
    Tree()                             //构造函数，初始化；
    {
        cnt = 0;
        for(int i=0; i<26; i++)
            next[i] = NULL;
    }
};
Tree *root;         // 树根；
char s[1020][25];

void creattree(char *str)     //创建树；
{

    int len = strlen(str);
    Tree *p = root;
    Tree *q;
    for(int i=0; i<len; i++)
    {
        int id = str[i] - 'a';     //id为根节点的第几个孩子；
                                   //这里不能直接p = p->next[id]; 必须新创一个Tree先赋给p->next[id];
        if(p->next[id] == NULL){
            q = new Tree;
            q->cnt = 1;
            p ->next[id]= q;
        }
        else
            p->next[id]->cnt++;     //记录有几个单词路过此节点；
        p = p->next[id];
    }

}

int findcnt(char *str)            //找到节点数为1的点，说明到该节点单词唯一；
{
    int len = strlen(str);
    Tree *t = root;
    int i;
    for(i=0; i<len; i++)
    {
        int id = str[i] - 'a';
        t = t->next[id];
        if(t->cnt == 1)
            return ++i;           //加上第一个节点：当字母为“a”时，i=0，所以要++；
    }
    return i;                     //当i=len时，退出，说明只能完全打印该单词；
}

int main()
{
    int n=0;
    root = new Tree;
    while(scanf("%s",s[n])!=EOF)   //输入结束时，enter->Ctrl+z->enter方可结束；
    {
        getchar();
        creattree(s[n]);
        n++;
    }
    for(int i=0; i<n; i++)
    {
        cout << s[i] << " ";
        int len = findcnt(s[i]);   //返回需要打印单词的长度；
        for(int j=0; j<len; j++)
            cout << s[i][j];
        cout <<endl;
    }
    return 0;
}

不难，只是比较复杂。