Shortest Prefixes POJ - 2001 (Trie树基本运用)

最新推荐文章于 2019-04-17 17:35:59 发布

原创最新推荐文章于 2019-04-17 17:35:59 发布 · 186 阅读

0 ·

CC 4.0 BY-SA版权

字符串---Tire树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用Trie树实现的算法，该算法能够为给定词典中的每个单词找到一个唯一的前缀，使得这些前缀可以区分每一个单词。通过构建Trie树并遍历每个单词来寻找其唯一的前缀标识符。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意:

给定多个单词组成的词典, 输出每个单词对应的前缀, 使得每个前缀唯一.

如果不存在就输出单词本身. 也就是:找出一个前缀,该前缀能唯一表示该字符串.

分析:

建立Trie树,每个节点的V代表有多少个字符串路径上含有它.

Find函数里查找每个单词的前缀,知道v值==1返回即可.

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

const int maxn = 1e7+11;

struct node
{
    int ch[maxn][26];
    int v[maxn];
    int sz;
    void init()
    {
        sz=1;
        memset(ch[0],0,sizeof ch[0]);
        v[0]=0;
    }
    void insert(char *s)
    {
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            if(!ch[u][id])
            {
                ch[u][id]=sz;
                memset(ch[sz],0,sizeof ch[sz]);
                v[sz]=0;
                sz++;
            }
            u=ch[u][id];
            v[u]++;
        }
    }
    void find(char *s)
    {
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            u=ch[u][id];
            printf("%c",s[i]);
            if(v[u]==1)//该字符已经能唯一标示该单词
            return;
        }
    }
}trie;

int main()
{
    char word[1111][33];
    trie.init();
    int cnt=0;
    while(scanf("%s",word[cnt])==1)
    {
        trie.insert(word[cnt++]);
    }
    for(int i=0;i<cnt;i++)
    {
        printf("%s ",word[i]);
        trie.find(word[i]);
        puts("");
    }
}