Trie树 Hihocoder 1014 Trie树

题目链接：http://hihocoder.com/problemset/problem/1014

题意：给一些字符串作为字典，再给一些前缀问前缀为这个的有多少个单词。

思路：Trie树模板。

建立的思想不难，用指针指向下一个字符。因为字符长度不定，而且如果是26字母的话就会过长，所以用指针来做最好。那么问题来了。

首先声明一个Trie的时候必须要把他里面标记置为0，把所有指向儿子的Trie指针置为NULL。插入时，使用指针*p指向插入字符串的首地址a，方便接下来开辟空间；用*cur=root来把cur指向root，方便往下遍历。如果当前son为NULL，使用cur->son[*p-’a’] = new Trie进行动态分配。查询同。

源码：

#include <cstdio>

#include <cmath>

#include <cstring>

#include <string>

#include <algorithm>

using namespace std;

int const MAXNODE = 26;

struct Trie

{

    int val;

    Trie *son[MAXNODE];

    Trie(){

        val = 0;

        for(int i=0; i<MAXNODE; i++)

            son[i] = NULL;

    }

};

void insert(Trie *root,char *a)

{

    if(root == NULL || *a=='\0')

        return;

    Trie *cur = root;

    char *p = a;

    for(int i=0; i<strlen(a); i++){

        if(cur->son[*p-'a']==NULL)

            cur->son[*p-'a'] = new Trie;

        cur = cur->son[*p-'a'];

        cur->val++;

        p++;

    }

}

int query(Trie *root,char *a)

{

    if(root==NULL || *a=='\0')

        return 0;

    Trie *cur = root;

    char *p = a;

    for(int i=0; i<strlen(a); i++){

        if(cur->son[*p-'a'] == NULL)

            return 0;

        cur = cur->son[*p-'a'];

        p++;

    }

    return cur->val;

}

int main()

{

    int n,i;

    char a[20];

    scanf("%d",&n);

    Trie *root = new Trie;

    while(n--){

        scanf("%s",a);

        insert(root, a);

    }

    scanf("%d",&n);

    for(int i=0; i<n; i++){

        scanf("%s",a);

//        getchar();

        printf("%d\n",query(root,a));

    }

    delete root;

    return 0;

}