hdu1251(字典树)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1251

思路:算是字典树模板吧, 好像没什么思路可言......

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define lc (d<<1)
#define rc (d<<1|1)
#define P pair
#define pi acos(-1)
struct as{
int v;
int ne[26];
void init()
{
    v=0;
    REW(ne,-1);
}}tr[500000];
int pos;
char a[15];
void add(char s[])
{
    int i=0,j;
    int p=0;
    while(s[i])
    {
        int ant=s[i]-'a';
        if(tr[p].ne[ant]==-1)
        {
            tr[pos].init();
            tr[p].ne[ant]=pos++;
        }
        p=tr[p].ne[ant];
        tr[p].v++;
        i++;
    }
}
void query(char s[])
{
    int i=0,j;
    int p=0;
    while(s[i])
    {
        int ant=s[i]-'a';
        if(tr[p].ne[ant]==-1)
        {
            printf("0\n");
            return;
        }
        p=tr[p].ne[ant];
        i++;
    }
    printf("%d\n",tr[p].v);
}
int main()
{
    cin.tie(0);
    pos=1;
    tr[0].init();
    gets(a);
    while(a[0]!='\0')
    {
        add(a);
        gets(a);
    }
    while(ss(a)!=EOF)
    {
        query(a);
    }
    return 0;
}

你可能感兴趣的:(字典树)