SDUT-2828 字典树

原题链接：sdut-2828 字典树

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct node
{
    int data;//记录出现了几次，这个字符串
    struct node *next[26];//26个字母，二十六个孩子
}ST;
ST a[1000000];//静态内存
int top;
ST *creat_kong()
{
    int i;
    ST *root = &a[top++];
    root->data = 0;
    for(i = 0; i < 26; i++)
    root->next[i] = NULL;
    return root;
}
ST *insert(ST *root, char str[])
{
    ST *p = root;
    int i, t;
    for(i = 0; str[i]; i++)//将字符串的字符一个一个放入字典树的结点中
    {
        t = str[i] - 'a';//下标，0到25分别对应这二十六个字母
        if(!p->next[t]) p->next[t] = creat_kong();//如果对应下标的孩子结点是空，创造一个
        p = p->next[t];//否则就继续
    }
    p->data++;//让字符串中最后一个字符，对应的结点的data值++，记录该字符出现了几次
    return root;
}
int find(ST *root, char str[])
{
    ST *p = root;
    int i, t;
    for(i = 0; str[i]; i++)
    {
        t = str[i] - 'a';//下标，0到25分别对应这二十六个字母
        if(!p->next[t]) return 0;//如果是空，代表没有一样的字符串return 0，也就是没有找到
        p = p->next[t];//否则继续
    }
    return p->data;//字符串循环成功走完，代表有它的母串，或者一样的，如果是母串就返回0，如果一样就返回出现了几次
}
int main()
{

    int n, m, i;
    char str[12];
    while(~scanf("%d %d", &n, &m))
    {
        top = 0;//用来控制使用静态内存
        ST *root = creat_kong();//先建好字典树的根，空的字典树
        if(n == 0 && m == 0) break;
        for(i = 0; i < n; i++)
        {
            scanf("%s", str);//输入字符串
            root = insert(root, str);//构建字典树
        }
        while(m--)//m次查询
        {
            scanf("%s", str);//输入查询的字符串
            if(find(root, str)) printf("Yes\n");//如果不是0
            else printf("No\n");
        }
    }
    return 0;
}