某Trie树的简单例题

初学Trie树!解决字符串前缀问题的法宝!
T1:poj2001:Shortest Prefixes
题意:给一堆字符串,问每个字符串最短的不与其他字符串前缀重复的前缀(好奇怪的表述QAQ
貌似“字符串前缀”之类的先考虑Trie树就好了。。。
思路:建Trie树,每路过一个点该点cnt++;最后查询时发现cnt==1的点则为目标,如果找到最后还不行,就直接输出。
代码:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#define cls(x) memset(x,0,sizeof x)
const int maxn = 1010;
using namespace std;
struct trie
{
    int next[27];
    int cnt;
}node[maxn<<5];
char c[maxn][25];
int tot;
void add(char* str)
{
    int rt=0;
    for(int i=0;str[i]!='\0';i++)
    {
        int x=str[i]-'a';
        if(node[rt].next[x])
            rt=node[rt].next[x];
        else
        {
            node[rt].next[x]=++tot;
            rt=tot;
        }
        node[rt].cnt++;
    }
}
void query(char *str)
{
    int rt=0;
    char b[100];cls(b);
    for(int i=0;str[i]!='\0';i++)
    {
        int x=str[i]-'a';
        if(node[rt].cnt==1)
        {
            printf("%s\n",b);
            return ;
        }
        rt=node[rt].next[x];
        b[i]=str[i];   //Trie树节点是空的,边上是字符,这句话要放在最后而不是前面
    }
    printf("%s\n",str);
    return ;
}
int main()
{
    int amt=0;
    while(~scanf("%s",c[++amt])) add(c[amt]);
    amt--;  // 不要问这是什么。。。EOF时amt也++了
    for(int i=1;i<=amt;i++)
    {
        printf("%s ",c[i]);
        query(c[i]);
    }
    return 0;
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/Flere825/article/details/51477238
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞