POJ1451 - 字典树的变型

上个周末给学校新队员上课...很是紧张..周日那场想来现场coding..就是很简单的高精度加法...结果怯场了..还是狐狸大大上来救了我..好丢人~~囧..这一向状态一直怪怪的~~这道题起码卡了三天吧~~其实这三天也是招新各种事~~没多大心思来做题~~昨天晚上终于把这题搞定了~~

题意就是说给一个字典...并同时给出每个单词的出现次数...是想一下手机的输入法...然后按手机的输入一串按键~~每按一下按键输出一个推荐字串~~这个字串是在字典中次数最多的前缀..如果有两个相等的..输出字典序靠前的....如果输入一串按键，在字典里一个也没得.那就只能输出"MANUALLY"...其实感觉这题是很实用很现实的~~咱们平时用手机输入法就是有这个性质~~

这道题还是的字典树...要求的是前缀出现次数最多..每个点的定义:

struct node { int s[26],x,father,w; char c; }

这道题因为有多组数据..动态分配空间如果不释放又很浪费..释放又很耗时...所以采取了开辟静态空间再来链接的伪指针方式....s[ ] 则是记录起孩子是否能链接到哪个位置...x是记录有多少个单词会经过这个点...所以在构造字典树时..每经过一个点..这个点的x就++...father是记录字典树中每个点的父亲..这里是为了这道题后面的输出方便..w是记录当前点所在层数..同样也是为了后面的处理方便...c是当前点是哪个字符..同样也是为了后面的处理方便....s[ ],father,x,w,c在构造树时都能得到准确的值....

构造树的过程和我上一个日志类似..只是多加几个运算得到father,x,w,c而已了~~就直接看构造完树后怎么得到想要的结果..做的话用一个队列来维护..先是超级头结点0入内..可以想象一下...每一次读一个字符就往下做一层...而队列的目的就是看每一层还有多少点可以进行拓展.....每次读一个字符c..如果不是'1'就继续做...扫描所有还能符合条件的上一层的点..找到他们孩子中即在手机按键上是这个c-'0'又其存在...这种点先入队..然后判断其x值..找到所有孩子中满足条件的点..如果没有一个点能入队代表不存在符合输入信息的前缀了~~就要跳出输出"MANUALLY"...如果能找到最佳前缀...则输出..输出的话就从这个最佳的点开始用father往上走..直到走到father为0..就能得到这个前缀了..

自己在写的时候调了很多bug...整了不少时间才搞定...还是自己队字典树不大熟练吧...囧...

Program:

#include<iostream> #include<queue> #define MAXN 1002 using namespace std; struct node { int s[26],x,father,w; char c; }a[10001]; int t,T,w,m,i,p[MAXN],l,g,newdata; int turn[26]={2,2,2,3,3,3,4,4,4,5,5,5,6,6,6,7,7,7,7,8,8,8,9,9,9,9}; char s[MAXN][101],c; void find(int h,int k) { if (k>l) return; a[h].x+=p[i]; if (!a[h].s[s[i][k]-'a']) { g++; a[h].s[s[i][k]-'a']=g; a[g].father=h; a[g].w=k+1; a[g].c=s[i][k]; if (k==l) a[g].x=p[i]; else a[g].x=0; } find(a[h].s[s[i][k]-'a'],k+1); } queue<int> myqueue; void make() { int h,k,m,i,step=0; char s[111]; while (!myqueue.empty()) myqueue.pop(); myqueue.push(0); while (scanf("%c",&c)!=EOF) { if (c=='1') return; h=myqueue.front(); m=0; while (a[h].w==step) { for (i=0;i<26;i++) if ((turn[i]==c-'0')&&(a[h].s[i])) { myqueue.push(a[h].s[i]); if (a[a[h].s[i]].x>m) { k=a[h].s[i]; m=a[k].x; } } myqueue.pop(); if (myqueue.empty()) break; h=myqueue.front(); } if (!m) goto A; step++; s[step]='\0'; for (i=step-1;i>=0;i--) { s[i]=a[k].c; k=a[k].father; } puts(s); } A: printf("MANUALLY\n"); while (scanf("%c",&c)!=EOF) { if (c=='1') return; printf("MANUALLY\n"); } return; } int main() { scanf("%d",&T); for (t=1;t<=T;t++) { scanf("%d",&w); g=0; memset(a,0,sizeof(a)); p[0]=0; for (i=1;i<=w;i++) { scanf("%s%d",s[i],&p[i]); l=strlen(s[i])-1; find(0,0); } printf("Scenario #%d:\n",t); scanf("%d",&m); a[0].x=0; while (m--) { getchar(); make(); printf("\n"); } printf("\n"); } return 0; }