首页 > 代码库 > POJ 2503 字典树
POJ 2503 字典树
这题记得以前是我们周赛的题,然后用的是map,也暴过了。
因为这两天要给大一的讲字典树,所以练练几道的代码,以防给大一搞晕了……
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<map> #include<queue> #include<set> #include<cmath> #include<bitset> #define mem(a,b) memset(a,b,sizeof(a)) #define lson i<<1,l,mid #define rson i<<1|1,mid+1,r #define INF 510010 #define maxn 400010 using namespace std; typedef long long ll; typedef unsigned long long ull; int ch[maxn][27]; //节点编号 int sz; //字典树的节点个数 int val[maxn]; //节点的值,为1说明新增一个字符串 char str[maxn][11]; void init() //初始化 { sz=1; memset(ch,0,sizeof(ch)); memset(val,0,sizeof(val)); } void insert(char *s,char *ss) { int u=0,i,c,l=strlen(ss); for(i=0; i<l; i++) { c=ss[i]-'a'; if(!ch[u][c]) ch[u][c]=sz++; u=ch[u][c]; } val[u]=1; strcpy(str[u],s); } void query(char *s) { int u=0,i,c,l=strlen(s),flag=1; for(i=0; i<l; i++) { c=s[i]-'a'; if(!ch[u][c]) {flag=0;break;} u=ch[u][c]; } if(val[u]&&flag) printf("%s\n",str[u]); else puts("eh"); } int main() { char s[22],s1[12],s2[12]; init(); while(gets(s),s[0]!='\0') { int i,len=strlen(s),j=0,k=0,flag=1; for(i=0;i<len;i++) { if(s[i]!=' '&&flag) s1[j++]=s[i]; else if(s[i]!=' '&&!flag) s2[k++]=s[i]; else flag=0; } s1[j]='\0';s2[k]='\0'; insert(s1,s2); } while(gets(s)) query(s); return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。