统计难题(trie树)
2015-10-25 19:16
357 查看
统计难题
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)Total Submission(s): 26206 Accepted Submission(s): 10637
[align=left]Problem Description[/align]
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
[align=left]Input[/align]
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
[align=left]Output[/align]
对于每个提问,给出以该字符串为前缀的单词的数量.
[align=left]Sample Input[/align]
banana
band
bee
absolute
acm
ba
b
band
abc
[align=left]Sample Output[/align]
2 3 1 0
题解:找以串s为前缀的字母个数,trie树,第一道,数组开小了,显示超时,错了半天;
字典树其实就是把下个字母当作过程,来往下查找,类似于哈希的思路;word代表当前前缀的个数
代码:
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #include<cmath> #include<vector> #define mem(x,y) memset(x,y,sizeof(x)) using namespace std; typedef long long LL; const int MAXN=1e6+10; char str[MAXN]; int ch[MAXN][30]; int word[MAXN]; int sz; void insert(char *s){ int len=strlen(s); int k=0,j; for(int i=0;i<len;i++){ j=s[i]-'a'; if(!ch[k][j]){ mem(ch[sz],0); ch[k][j]=sz++; } k=ch[k][j]; word[k]++; } } int find(char *s){ int len=strlen(s); int k=0,j; for(int i=0;i<len;i++){ j=s[i]-'a'; if(!ch[k][j])return 0; k=ch[k][j]; } return word[k]; } int main(){ sz=1; mem(ch[0],0);mem(word,0); while(gets(str),str[0]){ insert(str); } while(~scanf("%s",str)){ printf("%d\n",find(str)); } return 0; }
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> using namespace std; typedef long long LL; #define mem(x,y) memset(x,y,sizeof(x)) #define SI(x) scanf("%d",&x) #define PI(x) printf("%d",x) #define P_ printf(" ") const int INF=0x3f3f3f3f; const int MAXN=1000010; int ch[MAXN][30]; int word[MAXN]; int sz; void insert(char *s){ int k=0,j; for(int i=0;s[i];i++){ j=s[i]-'a'; if(!ch[k][j]){ mem(ch[sz],0); ch[k][j]=sz++; } k=ch[k][j];//注意不能要else word[k]++; } } int find(char *s){ int k=0,j; for(int i=0;s[i];i++){ j=s[i]-'a'; if(ch[k][j])k=ch[k][j]; else return 0; } return word[k]; } int main(){ char s[15]; sz=1; mem(word,0);mem(ch[0],0); while(gets(s)){ if(s[0]!='\0')insert(s); else break; } while(gets(s)){ printf("%d\n",find(s)); } return 0; }
相关文章推荐
- Spark基本概念
- android 54 播放音视频
- ARM9开发板FL2440移植Linux-3.0内核————添加USB驱动
- 关于 HTML5本地存储详解 (转载学习)
- LNMP一键安装包 V1.1 通告
- Python:使用threading模块实现多线程(转)
- 第四次实验——栈和队列(二)
- 信息安全系统设计基础第七周学习总结
- HTML问答
- CSS
- 猜数字小游戏
- 递归实现 1+2+3+...100 php和java
- Android工具Uitls
- [Android基础]布局属性
- hibernate的几个重要的接口
- 课堂讨论电子版
- 快速模幂运算
- 【知识点整理(java)】Java Thread简单使用注意点汇总(三)
- SVN服务器简单安装
- 第八周-项目4-字符串加密(有bug)