HDOJ2846解题报告【字典树】

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=2846

题目概述：

　　给你P个单词以及Q个询问，对于每个询问，回答一个整数，表示询问的字符串是多少个所给的单词的子串。

大致思路：

　　很容易看出是一个字典树的题，因为只要是子串都满足题意，所以在插入的时候需要多插入一些，例如：

　　　　对于单词abcd，则需要插入字符串abcd，bcd，cd，d；

　　然后插入的时候顺便统计一下个数，这样查询起来就当做正常的字典树查询就可以了。

　　需要注意的是，对于同一个字符串，插入子串时如果没有新建节点是不要使该节点统计的个数增加的。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <ctime>
#include <map>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;

#define sacnf scanf
#define scnaf scanf
#define maxn 1000000
#define maxm 26
#define inf 1061109567
#define Eps 0.001
const double PI=acos(-1.0);
#define mod 1000000007
#define MAXNUM 10000
void Swap(int &a,int &b) {int t=a;a=b;b=t;}
int Abs(int x) {return (x<0)?-x:x;}
typedef long long ll;

struct node
{
    int num;int cnt;
    int next[maxm];
} tire[maxn];

int c=1;
char s[maxm];

void Insert(int root,char *s,int n)
{
    while(*s!='')
    {
        int t=*s-'a';
        if(tire[root].next[t]==-1)
        {
            int temp=c++;
            for(int i=0;i<maxm;i++) tire[temp].next[i]=-1;
            tire[temp].cnt=0;tire[root].num=n;
            tire[root].next[t]=temp;
        }
        root=tire[root].next[t];
        if(tire[root].num!=n) tire[root].cnt++;      //不需要增加个数的情况
        tire[root].num=n;s++;
    }
}

int found(int root,char *s)
{
    while(*s!='')
    {
        int t=*s-'a';
        if(tire[root].next[t]==-1) return 0;
        root=tire[root].next[t];s++;
    }
    return tire[root].cnt;
}

int main()
{
    //freopen("data.in","r",stdin);
    //freopen("data.out","w",stdout);
    //clock_t st=clock();
    int root=0;
    for(int i=0;i<maxm;i++) tire[root].next[i]=-1;
    tire[root].num=0;tire[root].cnt=0;
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        int len=strlen(s);
        for(int t=0;t<len;t++)
            Insert(root,s+t,i);
    }
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",s);
        printf("%d
",found(root,s));
    }
    //clock_t ed=clock();
    //printf("

Time Used : %.5lf Ms.
",(double)(ed-st)/CLOCKS_PER_SEC);
    return 0;
}