【杭电】[1251]统计难题

统计难题

Problem Description

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

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束. 

Output

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

 

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2 3 1 0

字典树模板题，对每个节点记录cnt

最后一个字母的cnt便是以s为前缀的单词个数

注意释放内存

杭电G++一直爆内存……C++可以AC

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct Trie {
	int cnt;
	Trie *next[26];
}*root;
void Insert(char *s) {
	Trie *p,*t;
	int l=strlen(s);
	p=root;
	for(int i=0; i<l; i++) {
		if (p->next[s[i]-'a']==NULL) {
			t=new Trie;
			t->cnt=1;
			for (int j=0; j<26; j++)
				t->next[j]=NULL;
			p->next[s[i]-'a']=t;
		} else
			p->next[s[i]-'a']->cnt++;
		p=p->next[s[i]-'a'];
	}
}
int Search(char *s) {
	Trie *p;
	int l=strlen(s);
	p=root;
	for(int i=0; i<l; i++) {
		if (p->next[s[i]-'a']==NULL)
			return 0;
		p=p->next[s[i]-'a'];
	}
	return p->cnt;
}
void Release(Trie *p) {
	if (p==NULL)
		return ;
	for (int i=0; i<26; i++)
		if (p->next[i] != NULL)
			Release(p->next[i]);
	free(p);
	root = NULL;
	return ;
}
void Init() {
	root=new Trie;
	root->cnt=0;
	for (int i=0; i<26; i++)
		root->next[i]=NULL;
}
int main () {
	char str[15];
	Init();
	while (gets(str),strcmp(str,"")!=0)
		Insert(str);
	while (scanf ("%s",str)!=EOF)
		printf ("%d\n",Search(str));
	Release(root);
	return 0;
}

题目地址:【杭电】[1251]统计难题