HDU 2846 Repository（字典树）

 

题目大意

 

给了 P(1<=P<=10000) 件商品的名字(名字长度不超过20)，现在有 Q(1<=Q<=100000) 个查询，每个询问是询问一个字符串，问的是所有的商品名字中，包含这个字串的名字的数量

 

做法分析

 

做法比较裸，先把所有的商品名字以及这些名字的后缀建立一棵字典树，利用后缀的思想

那么，很容易想到：一个给定的查询字符串，一定是这些所有的后缀的一个前缀

现在等于是问：以这个字符串为前缀的不同后缀的个数

 

在建立字典树的时候，对于数中的每个节点，都需要一个标记 cnt：表示从根节点到当前节点，路径上经过的字符形成的字符串，被多少个不同的单词的后缀覆盖过

举个例子：

        有两个单词：aba 和 bc

        他们的所有后缀是： aba(1) ba(1) a(1) bc(2) c(2) （括号中的数字表示这个后缀属于哪个单词）

        那么，建立好的字典树就是：

怎么建树，其实很简单，大家稍微的想想就出来了

 

参考代码

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

struct Tire
{
    struct Node
    {
        Node *next[26];
        int cnt, flag;
        Node()
        {
            for(int i=0; i<26; i++) next[i]=NULL;
            cnt=0, flag=-1;
        }
    };

    Node *root;
    Tire()
    {
        root=new Node();
    }

    void Insert(char buff[], int s, int fg)
    {
        Node *u=root;
        for(int i=s; buff[i]; i++)
        {
            int v=buff[i]-'a';
            if(u->next[v]==NULL) u->next[v]=new Node();
            u=u->next[v];
            if(u->flag!=fg)
            {
                u->cnt++;
                u->flag=fg;
            }
        }
    }

    int Find(char buff[])
    {
        Node *u=root;
        for(int i=0; buff[i]; i++)
        {
            int v=buff[i]-'a';
            if(u->next[v]==NULL) return 0;
            u=u->next[v];
        }
        return u->cnt;
    }
} tree;

char s[30];
int n;

int main()
{
    scanf("%d", &n);
    for(int i=0; i<n; i++)
    {
        scanf("%s", s);
        int t=(int)strlen(s);
        for(int j=0; j<t; j++) tree.Insert(s, j, i);
    }
    scanf("%d", &n);
    for(int i=0; i<n; i++)
    {
        scanf("%s", s);
        printf("%d\n", tree.Find(s));
    }
    return 0;
}

AC通道

HDU 2846 Repository