字典树的题目。首先根据输入的单词建一棵字典树。然后每输入一个单词,就查询一次,若查到了,则结束。否则,则根据题目的意思,对该单词做三种操作(插入,删除,替换),然后再查询一次,若查到了则记录下来,最后排序输出。
再说下本题要注意的问题吧,对于插入和删除操作,可能会有重复的。比如字典树中已有单词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;
}