0
点赞
收藏
分享

微信扫一扫

集训队专题(1)1002 统计难题


统计难题

Time Limit : 4000/2000ms (Java/Other)   Memory Limit : 131070/65535K (Java/Other)
Total Submission(s) : 92   Accepted Submission(s) : 20


Problem Description


Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input


输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 注意:本题只有一组测试数据,处理到文件结束.

Output


对于每个提问,给出以该字符串为前缀的单词的数量.


Sample Input


banana band bee absolute acm ba b band abc

Sample Output


2 3 1 0

 

Author


Ignatius.L


此题也是基础的字典树问题,在这里,我用两种方法给大家呈现

第一种:指针实现。

指针实现的字典树最大的优点就是从代码上来看,形象且浅显易懂,但由于大量的空指针存在,可能从内存的角度讲,不是很好(在OJ problem里此题交指针是能过的,但在Web DIV里同样交此题就会出现爆内存的情况)


#include <iostream>//指针实现
#include <cstdio>
#include <cstring>
using namespace std;
char str[11];
struct node
{
node *next[26];
int cnt;
node()
{//构造函数,创建结点时自动执行
cnt=0;
for(int i=0; i<26; i++)
next[i] = NULL;//将该节点的下面的26个节点初始化为空
}
};
void insert(node *p,char *str)
{
for(int i=0; str[i]; i++)
{
int t=str[i]-'a';
if(p->next[t] == NULL)
p->next[t] = new node;//如果节点不存在,那么先new一个
p = p->next[t];
p->cnt++;
}
}
int find(node *p,char *str)
{
for(int i=0; str[i]; i++)
{
int t=str[i]-'a';
p = p->next[t];
if(!p) return 0;
}
return p->cnt;
}
int main()
{
node *root=new node();
while(gets(str)&&strlen(str))
insert(root,str);
while(gets(str))
{
int ans=find(root,str);
printf("%d\n",ans);
}
return 0;
}


第二种:数组模拟实现。

数组模拟实现的优点自然是从内存角度讲比较节省空间,但是初学者可能看到数组模拟的代码比较茫然,不知道是如何实现的,请大家多点耐心,务必学会这种方法。


#include <cstdio>//数组模拟实现 
#include <cstring>
#define N 5000000
using namespace std;
int tree[N][26],cnt[N];
char str[20];
int tot;
void insertTrie(char *str)
{
int len=strlen(str);
int now=0;
for(int i=0; i<len; i++)
{
int w=str[i]-'a';
if(tree[now][w] == 0) tree[now][w] = ++tot;
now = tree[now][w];
cnt[now]++;
}
}
int searchTrie(char *str)
{
int len=strlen(str);
int now=0;
for(int i=0; i<len; i++)
{
int w=str[i]-'a';
if(tree[now][w] == 0) return 0;
now = tree[now][w];
}
return cnt[now];
}
int main()
{
while(gets(str)&&strcmp(str,"")!=0)
insertTrie(str);
while(gets(str))
printf("%d\n",searchTrie(str));
return 0;
}




举报

相关推荐

0 条评论