Spell checker poj1035

    字典树的题目。首先根据输入的单词建一棵字典树。然后每输入一个单词，就查询一次，若查到了，则结束。否则，则根据题目的意思，对该单词做三种操作（插入，删除，替换），然后再查询一次，若查到了则记录下来，最后排序输出。

再说下本题要注意的问题吧，对于插入和删除操作，可能会有重复的。比如字典树中已有单词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;
}