统计难题
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input banana
band
bee
absolute
acm
ba
b
band
abc
Sample Output 2
3
1
0
题意:给你一个单词表,再随便给你一个串以这个串为前缀的单词有多少少;
思路:建一颗字典数,有相同前缀的合并,每个节点中有前缀个数,查询时从上向下查询就行了;
失误:错了好几次,不是超内存就是非法访问,最后才知道用动态内存写,这一题要掌握动态内存(题目中没有说数据多少,大小时用)还有memset对结构体初始化0的条件。
代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
using namespace std;
typedef struct tree{
int num;
struct tree *son[26];
}Node;
Node *root;
void initnode(Node *p) {
int i=0;
for(i=0;i<26;++i) p->son[i]=NULL;//结构体中有数组 不能用memset()
p->num=0;
}
void insert(char *str)//建树
{
int i=0; Node *p1=root,*p2;
while(str[i]!='\0')
{
int pos=str[i]-'a';
if(p1->son[pos]==NULL)
{
p2=new Node;
initnode(p2);
p1->son[pos]=p2;//没赋值 非法访问内存
}
p1=p1->son[pos];
++p1->num;
++i;
}
}
int find(char str[])
{
int i=0,cnt=0; Node *p=root;
while(str[i]!='\0')
{
int pos=str[i]-'a';
if(p->son[pos]==NULL) return 0;
p=p->son[pos];
cnt=p->num;
++i;//缺省输出都是0
}
return cnt;
}
int main()
{
char str[22]; root=new Node;//缺省非法访问内存
initnode(root);
while(gets(str),strcmp(str,""))
{
insert(str);
}
while(~scanf("%s",str))
{
printf("%d\n",find(str));
}
return 0;
}