BZOJ 1819: [JSOI]Word Query电子字典

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1819

分析：
暴力枚举需要添加、删除以及替换的字符，然后建立trie树进行判断是否存在该字符串，然后利用trie树存的信息进行判重

时间貌似有点慢。

329611

yejinru

1819

Accepted

21468 kb

1232 ms

C++/Edit

2193 B

2012-12-16 17:34:39

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;

const int kind = 26;
const int X = 25;

#define debug puts("here");

int m,n,tot;

struct trie{
	trie *p[kind];
	int is,id; //is 用于判重，id用于标记是否有单词在这个节点结束
	trie(){
		memset(p,NULL,sizeof(p));
		is = id = 0;
	}
};

void insert(trie *t,char *s){
	int id;
	for(int i=0;s[i];i++){
		id = s[i]-'a';
		if(t->p[id]==NULL)
			t->p[id] = new trie();
		t = t->p[id];
	}
	t->is = t->id = tot;
}

int has(trie *t,char *s){ //判断是否存在字符串s
	int c = 0;
	for(int i=0;s[i];i++){
		c = s[i]-'a';
		if(t->p[c]==NULL)
			return 0;
		t = t->p[c];
	}
	if(t->id)
		return 1;
	return 0;
}

int find(trie *t,char *s,int id){ //用于判断是否存在字符串s并且s没被统计过，id在这用于判重
	int c = 0;
	for(int i=0;s[i];i++){
		c = s[i]-'a';
		if(t->p[c]==NULL)
			return 0;
		t = t->p[c];
	}
	if(t->id==0||t->is==id) //不存在单词或者已经算过一次了
		return 0;
	t->is = id;
	return 1;
}

int main(){
	int n , m;
	while(cin>>n>>m){
		tot = 10000000;

		trie * root = new trie();
		char s[X],str[X];
		
		for(int i=0;i<n;i++){
			scanf("%s",s);
			insert(root,s);
		}
		
		while(m--){
			int ans = 0;
			scanf("%s",str);

			if(has(root,str)){
				puts("-1");
				continue;
			}

			tot --;

			int len = strlen(str);

			for(int i=0;i<len;i++){	// delete
				strcpy(s,str);
				
				for(int j=i+1;j<len;j++)
					s[j-1] = str[j];
				s[len-1] = '\0';
				
				ans += find(root,s,tot);
			}

			for(int i=0;i<len;i++){	// change
				for(int j=0;j<26;j++){
					if(str[i]==j+'a')
						continue;

					strcpy(s,str);
					s[i] = j+'a';
					
					ans += find(root,s,tot);
				}
			}

			for(int i=0;i<=len;i++){ // add
				for(int j=0;j<26;j++){
					
					s[i] = j+'a';
					for(int k=0;k<i;k++)
						s[k] = str[k];
					for(int k=i;k<len;k++)
						s[k+1] = str[k];
					s[len+1] = '\0';

					ans += find(root,s,tot);
				}
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}