poj 2418 Hardwood Species (trie树)

poj   2418   Hardwood Species 

http://poj.org/problem?id=2418

trie树+dfs

题意： 给你多个单词，问每个单词出现的频率。

方法：通过字典树，将所有单词放入树中，通过dfs遍历（题目要求按ASSIC码顺序输出单词及其频率），dfs可满足

注意：单词中不一定只出现26个英文字母，ASSIC码表共有256个字符

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <stack>
#include <queue>
#include <functional>
#include <vector>
#include <map>
using namespace std;
#define M 0x0f0f0f0f
#define min(a,b) (a>b?b:a)
#define max(a,b) (a>b?a:b)
int ji;
char st[40];
struct tree
{
    int count_;
    struct tree *next[256];
};

void insert_(struct tree *root,char *s)
{
    int i;
    if(root==NULL||*s=='\0')
        return ;
    struct tree *p=root;
    while(*s!='\0')
    {

        if(p->next[*s]==NULL)
        {
            struct tree *t=(struct tree *)malloc(sizeof(struct tree));
            for(i=0; i<256; i++)
                t->next[i]=NULL;
            t->count_=0;
            p->next[*s]=t;
            p=t;
        }
        else
        {
            p=p->next[*s];
        }
        s++;
    }
    p->count_++;
}
int j=0;
void dfs(struct tree *p,int j)
{
    int i;
    if(p->count_)
    {
        st[j]='\0';
        double m=(p->count_)*1.0/((ji)*1.0)*100;
        printf("%s %.4lf\n",st,m);
    }
    for(i=0; i<256; i++)
    {
        if(p->next[i]!=NULL)
        {
            st[j]=i;
            dfs(p->next[i],j+1);
        }
    }
}

int main()
{
    char a[10005],a1[10005];
    int i;
    struct tree *root=(struct tree *)malloc(sizeof(struct tree));
    for(i=0; i<256; i++)
    {
        root->next[i]=NULL;
    }
    root->count_=0;
    ji=0;
    while(gets(a)!=NULL)
    {
        ji++;
        insert_(root,a);
    }
    dfs(root,0);
    return 0;
}