hdu 1251 统计难题

字典树~ 一种很好的母串有木有含某子串的数据结构~这是一道模板题!当然入手要先做模板题啦~

很好理解的一种树~

#include <iostream>
#include <stdio.h>
using namespace std;
struct Trienode
{
    int cnt;
    Trienode* next[26];
};

void insert(Trienode *root,char *s)
{
    Trienode *p = root;
    int i = 0;
    while(s[i]!='\0')
    {
        //cout<<i<<endl;
        int k = s[i]-'a';
        if(p->next[k])
        {
            p->next[k]->cnt ++;
        }
        else
        {
            Trienode *tmp = new Trienode;
            for(int j = 0;j < 26;j ++)
            tmp->next[j] = NULL;tmp->cnt = 1;
            p->next[k] = tmp;
        }
        p = p -> next[k];
        i ++;
    }
}

int search(Trienode *root,char *s)
{
    int i,k;
    Trienode *p = root;
    i = 0;
    while(s[i]!='\0')
    {
        k = s[i] - 'a';
        if(p->next[k] == NULL) return 0;
        else p = p -> next[k];
        i ++;
    }
    return p->cnt;
}

int main()
{
    char s[20];
    Trienode *root = new Trienode;
    root ->cnt = 0;
    for(int j = 0;j < 26;j ++)
    root->next[j] = NULL;
    //cout<<"Das"<<endl;
    while(gets(s) && s[0])
    {
        insert(root , s);
    }

    while(gets(s))
    {
        printf("%d\n", search(root , s));
    }

    return    0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值