HDU 1251 统计难题 （Trie树模板题）

题目链接：点击打开链接

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2 3 1 0

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2 3 1 0

 

题意：Trie树模板题，统计出现次数，注意注释

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int maxn = 400000 + 20;
int tot;
char s[maxn];
int num[maxn] = {0};//前缀末尾编号为i的单词数量。
int trie[maxn][26];
//trie[i][j] = k表示编号为i的节点的第j个孩子（内容为j的孩子）是编号为k的节点
//编号为整棵树中的编号, 内容是值的编号,如小写字母的字典树中，'b'的编号为'b' - 'a'  = 1;

void insert(char s[]){//插入单词s
	int root = 0;//从整颗字典树的根节点开始找起，整颗字典树的根节点的编号为0
	int len = strlen(s);
	for(int i = 0; i < len; i++){
		int id = s[i] - 'a';
		if(!trie[root][id]) trie[root][id] = tot++;//如果之前没有从root到id过，自己创建一个。
		//tot为整棵树中的编号，为全局变量，编到几就是几
		root = trie[root][id];//无论从root到id的边是否之前存在，下一步都要往下走
		num[root]++;//每次沿字典树输入一个单词，都让相应编号的num++，方便输出以某字符串为前缀的单词数
	}
}

int find(char s[]){
	//前缀查找函数，注意：查找前缀是否存在与查找单词是否存在不同！
    //如：空字典树中插入apple , applese 存在前缀app,但是不存在单词app!
	//如果要查找单词，对应的insert函数要加入一个bool的v[i]标志：编号i的节点是单词结束的标志，find函数怎么改你懂的
	int root = 0;//依然是从根节点开始查找
	int len = strlen(s);
	for(int i = 0; i < len; i++){
		int id = s[i] - 'a';
		if(!trie[root][id]) return 0;
		root = trie[root][id];
	}
	return num[root];

}

int main(){

	memset(trie, 0 , sizeof(trie));//初始化很重要
	tot = 1; //上面insert中为  tot++ , 所以这里应该初始化为1
	while(gets(s)){//只有gets才能在遇到回车的时候返回长度为0的空串，gets同时也能记录空格
		if(strcmp(s,"")==0) break;//学会此处的输入方法:若字符串长度为0或者输入的为空格,结束循环.
		//strcmp(str1, str2)  若str1 == str2返回0， 若前者字典序大，返回正数， 否则返回负数
		insert(s);

	}
	//for(int i = 0; i < tot; i++)printf("num[%d]:%d\n", i, num[i]);
	//for(int i = 0; i < tot; i++)for(int j = 0; j < 26; j++) if(trie[i][j])printf("trie[%d][%c] = %d\n", i, j+'a', trie[i][j]);
	while(~scanf("%s",s)){
		printf("%d\n", find(s));
	}
    return 0;
}