2786: [JSOI]Word Query电子字典

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 3 Solved: 3
[Submit][Status][Web Board]

Description

人们在英文字典中查找某个单词的时候可能不知道该单词的完整拼法，而只知道该单词的一个错误的近似拼法，这时人们可能陷入困境，为了查找一个单词而浪费大量的时间。带有模糊查询功能的电子字典能够从一定程度上解决这一问题：用户只要输入一个字符串，电子字典就返回与该单词编辑距离最小的几个单词供用户选择。字符串a与字符串b的编辑距离是指：允许对a或b串进行下列“编辑”操作，将a变为b或b变为a，最少“编辑”次数即为距离。 删除串中某个位置的字母； 添加一个字母到串中某个位置； 替换串中某一位置的一个字母为另一个字母； JSOI团队正在开发一款电子字典，你需要帮助团队实现一个用于模糊查询功能的计数部件：对于一个待查询字符串，如果它是单词，则返回-1；如果它不是单词，则返回字典中有多少个单词与它的编辑距离为1。

Input

第一行包含两个正整数N (N < = 10,000)和M (M < = 10,000)。接下来的N行，每行一个字符串，第i + 1行为单词Wi。单词长度在1至20之间。再接下来M行，每行一个字符串，第i + N + 1表示一个待查字符串Qi。待查字符串长度在1至20之间。Wi和Qi均由小写字母构成，文件中不包含多余空格。所有单词互不相同，但是查询字符串可能有重复。

Output

输出应包括M行，第i行为一个整数Xi。Xi = -1表示Qi为字典中的单词；否则Xi表示与Qi编辑距离为1的单词的个数。

Sample Input

4 3
abcd
abcde
aabc
abced
abcd
abc
abcdd

Sample Output

-1
2
3
//abcd在单词表中出现过；abc与单词abcd、aabc的编辑距离都是1；abcdd与单词abcd、abcde、abced的编辑距离都是1。

HINT

Source

题解：

　　trie树+dfs乱搞出奇迹。。。。。。

#include<iostream>
#include<cstring>
#include<string> 
#include<cstdio>
#define maxn 200010
using namespace std;
int n,m,ans,len,top,tot;
int fail[maxn],sum[maxn],lis[maxn],son[maxn][26];
bool bo;
bool vis[maxn];
char s[maxn];
void insert()
{
    scanf("%s",s+1); int p=0;
    for (int i=1; s[i]; p=son[p][s[i]-'a'],i++)
        if (!son[p][s[i]-'a']) son[p][s[i]-'a']=++tot;
    sum[p]=1; 
}
void dfs(int x,int y, int z)  
{
    if (x==len && sum[y] && !z) {bo=false; return; }
    if (x==len && sum[y] && z) { if (!vis[y]) {vis[y]=1; lis[++top]=y; ans++;}return;}
    if (x<len && !z) dfs(x+1,y,1);//加单词 
    if (!z)
    {
        for (int i=0; i<26; i++)
            if (son[y][i]) { dfs(x,son[y][i],1); if (i!=s[x+1]-'a') dfs(x+1,son[y][i],1);}
    }
    //if (x==len) return ;
        if (son[y][s[x+1]-'a']) dfs(x+1,son[y][s[x+1]-'a'],z);
     
}
void work()
{
    while (top) vis[lis[top--]]=0;
 
    scanf("%s",s+1); len=strlen(s+1); ans=0; bo=true;
    dfs(0,0,0);
    if (bo==false) printf("-1
"); else printf("%d
",ans);
} 
int main()
{
    cin>>n>>m; top=tot=0;
    memset(vis,false,sizeof(vis));
    for (int i=1; i<=n; i++) insert();
    for (int i=1; i<=m; i++) work();
}

View Code