某Trie树的简单例题

来源：互联网发布：ajax.js 代码下载编辑：程序博客网时间：2024/06/08 10:20

初学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;}

1 0