hdu 1251 統計難題 (字典樹)

題目鏈接:   hdu 1251

題目大意:   給出單詞的詞典，然後有限次詢問

                  每次詢問給出的字符在詞典中作爲前綴的次數

解題思路:   建立詞典的字典樹

                  用w標記此結點在建樹過程中訪問的次數，每經過一次就+1

                  查詢時把查詢的字符遍歷字典樹，遍歷最後結點的w值既是答案

代碼:

#include <stdio.h>
#include <string.h>
#include <string.h>
#define MAX 1000000
struct snode{
    char ch1;
    int next;   //第二種寫法
}Tree[MAX];
char ch[20];
int pd,Index,num[MAX],pre[MAX];

void Add_edge(int Star,char ch2)   //建立有向圖
{
    Tree[Index].ch1=ch2,Tree[Index].next=pre[Star];
    num[Index]++;
    pre[Star]=Index++;
}

void DFS(int Star,int len,int Tlen)  //DFS插入結點
{
    int i;
    if(len>Tlen)
        return ;
    for(i=pre[Star];i!=-1;i=Tree[i].next)
    {
        if(Tree[i].ch1==ch[len-1])
        {
            num[i]++;
            if(len<Tlen)
            {
                DFS(i,len+1,Tlen);
            }
            return ;
        }
    }
    Add_edge(Star,ch[len-1]);
    DFS(Index-1,len+1,Tlen);
}

void DFS2(int Star,int len,int Tlen)   //DFS查詢w值
{
    int i;
    if(len>Tlen)
        return ;
    for(i=pre[Star];i!=-1;i=Tree[i].next)
    {
        if(Tree[i].ch1==ch[len-1])
        {
            if(len==Tlen)
            {
                pd=num[i];  //num[ ]是w值
                return ;
            }
            if(len<Tlen)
            {
                DFS2(i,len+1,Tlen);
            }
            return ;
        }
    }
    
}

int main()
{
    int n=0,i;
    Index=1;
    memset(num,0,sizeof(num));
    memset(pre,-1,sizeof(pre));
    memset(Tree,0,sizeof(Tree));
    Tree[0].ch1='\0';
    Tree[0].next=-1;
    while(gets(ch))    //輸入詞典
    {
        if(ch[0]=='\0')
            break;
        DFS(0,1,strlen(ch));
    }
    while(gets(ch))    //查詢
    {
        pd=0;
        DFS2(0,1,strlen(ch));
        printf("%d\n",pd);
    }
    return 0;
}