字典树的题目。首先根据输入的单词建一棵字典树。然后每输入一个单词,就查询一次,若查到了,则结束。否则,则根据题目的意思,对该单词做三种操作(插入,删除,替换),然后再查询一次,若查到了则记录下来,最后排序输出。
再说下本题要注意的问题吧,对于插入和删除操作,可能会有重复的。比如字典树中已有单词acm,现在输入单词acmm,显然删除第一个m和第二个m得到的单词是一样的,所以这就导致重复,因此对最后得到的所有单词,排序后还要去重,切记!!切记!!
//WA到死,最后发现增加和删除可能有重复的单词,我还是太年轻了 #include<iostream> #include<cstring> #include<algorithm> #include<cstdio> using namespace std; struct node { int order;//单词的出现顺序 bool isTail; node *next[26]; }memory[1000000],*root; int cur,cnt;//cnt记录单词数 struct word//记录经过修改后能匹配的单词 { int order; char s[20]; bool operator<(const word &t)const { return order<t.order; } }match[1000]; int res;//匹配数量 node *build_node() { node *p=&memory[cur++]; p->isTail=false; for(int i=0;i<26;i++) p->next[i]=NULL; return p; } void insert_tree(char *s) { int len=strlen(s); node *p=root; for(int i=0;i<len;i++) { int index=s[i]-'a'; if(p->next[index]==NULL) p->next[index]=build_node(); p=p->next[index]; } p->isTail=true; p->order=++cnt; } int query(char *s)//若匹配返回单词编号,否则返回0 { int len,index; node *p=root; len=strlen(s); for(int i=0;i<len;i++) { index=s[i]-'a'; if(p->next[index]==NULL) return 0; p=p->next[index]; } return p->isTail?p->order:0; } int main() { char s[20],tmp[20]; int i,j,len; cur=cnt=0; root=build_node(); while(scanf("%s",s)) { if(s[0]=='#') break; insert_tree(s); } while(scanf("%s",s)) { if(s[0]=='#') break; if(query(s)) { printf("%s is correct\n",s); continue; } res=0; len=strlen(s); for(i=0;i<len;i++)//枚举删除字母的位置 { for(j=0;j<i;j++) tmp[j]=s[j]; for(;j<len;j++) tmp[j]=s[j+1]; if((match[res].order=query(tmp))) { strcpy(match[res].s,tmp); res++; } } for(i=0;i<len;i++)//枚举替换位置及字母 { strcpy(tmp,s); for(j=0;j<26;j++) { tmp[i]=j+97; if((match[res].order=query(tmp))) { strcpy(match[res].s,tmp); res++; } } } for(i=0;i<=len;i++) { for(j=0;j<i;j++) tmp[j]=s[j]; for(;j<=len;j++) tmp[j+1]=s[j]; for(j=0;j<26;j++) { tmp[i]=j+97; if((match[res].order=query(tmp))) { strcpy(match[res].s,tmp); res++; } } } sort(match,match+res); printf("%s:",s); //去重然后输出 for(i=0;i<res;i++) { if(strcmp(s,match[i].s)==0) continue; strcpy(s,match[i].s); printf(" %s",s); } putchar('\n'); } return 0; }