HDOJ1251解题报告【字典树】

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

题目概述：

　　给你一些单词和一些询问，对于每个询问求出所有单词中以询问为前缀的单词个数。

大致思路：

　　稍微修改一下Tire树就好了，对每个节点增加一个标记来记录到这个节点共有几个单词，这个题还需要注意用g++提交指针版的会MLE，所以需要用数组形式的Tire树。

　　年轻的我还是被输入给坑了……果然naive.

代码：

#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 maxn 1000010
#define inf 1061109567
#define Eps 0.001
#define PI 3.1415927
#define mod 9973
#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;

int tree[maxn][26];
int cnt[maxn];
int dir=1;

void Insert(char *s)
{
    int p=0,len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(!tree[p][s[i]-'a'])
        {
            tree[p][s[i]-'a']=dir;
            cnt[dir]++;dir++;
        }
        else cnt[tree[p][s[i]-'a']]++;
        p=tree[p][s[i]-'a'];
    }
}

int query(char *s)
{
    int p=0,len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(tree[p][s[i]-'a']!=0) p=tree[p][s[i]-'a'];
        else return 0;
    }
    return cnt[p];
}

int main()
{
    //freopen("data.in","r",stdin);
    //freopen("data.out","w",stdout);
    //clock_t st=clock();
    char s[20];
    memset(cnt,0,sizeof(cnt));
    while(gets(s))
    {
        if(strlen(s)==0) break;
        Insert(s);
    }
    while(gets(s))
    {
        printf("%d
",query(s));
    }
    //clock_t ed=clock();
    //printf("

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