hdu 1251+hdu 1671（字典树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

一开始我是直接用STL做的，唉。。。没有经验那。。。orz...然后毫无疑问地超时了。。。

看别人blog说是字典树，说实话，第一次听到这个。。。就立马学了一下。。发现挺简单的。。。嘻嘻。。。

#include<iostream>
#include<cstring>
const int N=26;
using namespace std;
struct Tire{
    int v;
    Tire *next[N];
};
Tire *root;

//创建字典树
void CreateTire(char str[]){
    int len=strlen(str);
    Tire *p=root,*q;
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(p->next[id]==NULL){
            q=(Tire *)malloc(sizeof(Tire));
            q->v=1;//一开始为1;
            for(int i=0;i<N;i++){
                q->next[i]=NULL;
            }
            p->next[id]=q;
            p=p->next[id];
        }else {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
}

int FindTire(char str[]){
    int len=strlen(str);
    Tire *p=root;
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(p->next[id]==NULL){
            return 0;
        }
        p=p->next[id];
    }
    return p->v;
}

int main(){
    char str[100];
    root=(Tire *)malloc(sizeof(Tire));
    for(int i=0;i<N;i++){
        root->next[i]=NULL;
    }
    while(gets(str)&&str[0]!='\0'){
        CreateTire(str);
    }
    while(~scanf("%s",str)){
        int ans=FindTire(str);
        printf("%d\n",ans);
    }
    return 0;
}

 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1671

值得一提的是，最后要释放动态申请的空间，不然就MLE。。。

#include<iostream>
#include<cstring>
const int N=10;
using namespace std;
struct Tire{
    int v;
    Tire *next[N];
};
Tire *root;

//创建字典树
void CreateTire(char str[]){
    int len=strlen(str);
    Tire *p=root,*q;
    for(int i=0;i<len;i++){
        int id=str[i]-'0';
        if(p->next[id]==NULL){
            q=(Tire *)malloc(sizeof(Tire));
            q->v=1;
            for(int i=0;i<N;i++){
                q->next[i]=NULL;
            }
            p->next[id]=q;
            p=p->next[id];
        }else {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
}

int FindTire(char str[]){
    int len=strlen(str);
    Tire *p=root;
    for(int i=0;i<len;i++){
        int id=str[i]-'0';
        if(p->next[id]==NULL||p->next[id]->v==1)
            return 0;
        p=p->next[id];
    }
    return 1;
}
//动态开辟空间要释放，不然就要MLE;
void DelTire(Tire *root){
    if(root==NULL)return ;
    for(int i=0;i<N;i++){
        if(root->next[i]!=NULL){
            DelTire(root->next[i]);
        }
    }
    free(root);
}

int main(){
    int n,_case;
    scanf("%d",&_case);
    while(_case--){
        scanf("%d",&n);
        char str[10010][11];
        root=(Tire *)malloc(sizeof(Tire));
        for(int i=0;i<N;i++){
            root->next[i]=NULL;
        }
        for(int i=0;i<n;i++){
            scanf("%s",str[i]);
            CreateTire(str[i]);
        }
        int flag=0;
        for(int i=0;i<n;i++){
             flag=FindTire(str[i]);
            if(flag){
                printf("NO\n");
                break;
            }
        }
        if(!flag)printf("YES\n");
        DelTire(root);
    }
    return 0;
}

附上一大牛的blog:http://www.wutianqi.com/?p=1359