（字典树模板）统计难题--hdu--1251

 

链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1251

 

 在自己敲了一遍后终于懂了，这不就用了链表的知识来建立了树，对！就是这样的，然后再查找

代码：

#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
using namespace std;

struct node
{
    int sum;
    node *next[26];
};

void BuildLibTree(node *head, char s[])
{
    node *p = head;

    for(int i=0; s[i]; i++)
    {
        int k = s[i]-'a';

        if(p->next[k] == 0)
            p->next[k] = new node();
        p = p->next[k];
        p->sum++;
    }
}
int Query(node *head, char s[])
{
    node *p = head;

    for(int i=0; s[i]; i++)
    {
        int k = s[i]-'a';

        if(p->next[k] == 0)
            return 0;
        p = p->next[k];
    }

    return p->sum;
}

int main()
{
    char s[20];
    node *head = new node();

    while(gets(s), s[0])
        BuildLibTree(head, s);
    while(scanf("%s", s) != EOF)
        printf("%d\n", Query(head, s));

    return 0;
}