题目链接：点击打开链接
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;
}