Hihocoder #1014 : Trie树 (字典数树统计前缀的出现次数 *【模板】 基于指针结构体实现 )

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>

using namespace std;

struct node
{
    int flag; //用来存储经过这个节点的路径（单词）
    struct node *next[26];
};

struct node *newnode() //这是要创建的整个字典树的根节点的开始
{
    int i;
    struct node *p;
    p=new struct node;

    for(i=0; i<26; i++ )
    {
        p->next[i] = NULL;
    }
    p->flag=0; //这是用来标记用的
    return p;
};

void insert(struct node *root, char *s )
{
    struct node *p=root;//此处必须要替换一下 目的是保证主函数里的p指针是头指针不变
    int len, i;
    int t;
    len=strlen(s);
    struct node *temp;
    for(i=0; i<len; i++)
    {
        t=s[i]-‘a‘;
        if(p->next[t]==NULL)
        {
            temp=newnode();
            temp->flag++;
            //printf("%d--", temp->flag );
            p->next[t]=temp;
            p=p->next[t];
        }
        else
        {
            p=p->next[t];
            p->flag++;
           // printf("%d**", p->flag );
        }
    }
    //printf("\n");
}


int Search(struct node *root, char *t )
{
    int i, tt;
    int len=strlen(t);
    struct node *p=root;
    int cnt;

    for(i=0; i<len; i++)
    {
        tt=t[i]-‘a‘;
        if(p->next[tt]!=NULL )
        {
            p=p->next[tt];
            cnt=p->flag;
            //printf("%d--%d**", p->flag, cnt );
        }
        else
            break;
    }
    if( i==len )
        return cnt;
    else
        return 0;
}

struct node *p;

int main()
{
    int n, m;
    scanf("%d", &n);
    int i, j;
    char s[20], t[20];
    p=newnode();

    for(i=0; i<n; i++)
    {
        scanf("%s", s);
        insert(p, s);
    }

    scanf("%d", &m);
    for(i=0; i<m; i++)
    {
        scanf("%s", t);
        int k = Search(p, t);
        printf("%d\n", k );
    }
    return 0;
}