map

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm

ba
b
band
abc
Sample Output
2
3
1
0

#include<stdio.h>
#include<iostream>
#include<string.h>

#define maxn 27
using namespace std;

struct Trie
{
    Trie *next[maxn];
    int times;
};


void build(Trie *root,char *str,int len)
{
    int i,j;
    Trie *p=root;
    Trie *q;
    //int len=strlen(str);
    for(i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]!=NULL)
        {
            p=p->next[id];
            p->times++;
            continue;
        }
        q=(Trie *)malloc(sizeof(Trie));
        q->times=1;
        for(j=0;j<maxn;j++)
        {
            q->next[j]=NULL;
        }
        p->next[id]=q;
        p=p->next[id];
    }

}

int findTrie(Trie *root,char *str)
{//查询字符串出现的次数
    int len=strlen(str);
    Trie *p=root;
    int times=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            times=0;
            return times;
        }
        p=p->next[id];
        if(i==len-1)
        {
            times=p->times;
            return times;
        }

    }
    return 0;

}
int main()
{
    Trie *root;
    int j;
    root=(Trie *)malloc(sizeof(Trie));
    for(j=0;j<maxn;j++)
    {
        root->next[j]=NULL;
    }   
    while(true)
    {
        char ch[10];
        int i=0;
        while(true)
        {
            ch[i]=getchar();
            if(ch[i]=='\n')break;       
            i++;
        }
        if(i==0)
            break;
        build(root,ch,i);
    }
    char ch[10];
    while(scanf("%s",&ch)!=EOF)
    {
        printf("%d\n",findTrie(root,ch));
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值