POJ 2001 Shortest Prefixes 【Trie树】

<题目链接>

题目大意：

找出能唯一标示一个字符串的最短前缀，如果找不出，就输出该字符串。

解题分析：

Trie树的简单应用，对于每个单词的插入，都在相应字符对应的节点 num 值+1 ，这样在查询的时候，如果遍历到num值为1的节点，就可以确定，该前缀能够唯一确定一个字符串，或者是一直遍历到NULL,这时直接输出它本身即可。

指针式Trie树：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N =1e3+10;
char word[N][26];
struct Node{
    int num;
    Node *next[26];
    Node(){
        num=0;
        for(int i=0;i<26;i++)
            next[i]=NULL;
    }
};
Node *root;
Node *now,*newnode;
void Insert(char *str){
    now=root;
    for(int i=0;i<strlen(str);i++){
        int to=str[i]-'a';
        if(now->next[to]==NULL){   //如果该节点为空
            newnode=new Node;
            ++(newnode->num);
            now->next[to]=newnode;
            now=now->next[to];
        }
        else{
            now=now->next[to];   
            ++(now->num);
        }
    }
}
void Search(char *str){
    char ans[26];
    now=root;
    for(int i=0;i<strlen(str);i++){
        int to=str[i]-'a';
        now=now->next[to];
        ans[i]=str[i];    //ans[]记录下前缀字符串
        ans[i+1]='\0';
        if(now->num==1){    //如果该节点只有一个对应的公共前缀，那么就是它本身的前缀，所以直接输出该节点的对应前缀即可
            printf("%s %s\n",str,ans);
            return;
        }
    }
    printf("%s %s\n",str,str);    
}
int main(){
    root=new Node;
    int cnt=0;
    while(gets(word[++cnt])){
        //if(!strlen(word[cnt]))break;
        Insert(word[cnt]);
    }
    for(int i=1;i<=cnt;i++)Search(word[i]);
    return 0;
}

 

数组式Trie树        转载于   >>>

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
char a[10001][31];
int tot,size;
int sz,t[300001][26],s[300001];
void insert(char ch[])
{
    int k,len=strlen(ch+1),now=0;
    for(int p=1;p<=len;p++)
    {
          k=ch[p]-'a';
          if(t[now][k]==0)t[now][k]=++sz;
        now=t[now][k];
        s[now]++;
    }
}
void ask(char ch[])
{
    int now=0,k,len=strlen(ch+1);
    for(int p=1;p<=len;p++)
    {
        if(s[now]==1)break;
        k=ch[p]-'a';
        printf("%c",ch[p]);
        now=t[now][k];
   }
}
int main()
{
    while(scanf("%s",a[++tot]+1)!=EOF)insert(a[tot]);
    for(int i=1;i<=tot;i++)
    {
        printf("%s ",a[i]+1);
        ask(a[i]);
        printf("\n");
    }
    return 0;
}

 

 

2018-10-29