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

1819: [JSOI]Word Query电子字典

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 1072  Solved: 361
[Submit][Status][Discuss]

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均由小写字母构成，文件中不包含多余空格。所有单词互不相同，但是查询字符串可能有重复。 提示：有50%的数据范围：N < = 1,000，M < = 1,000。

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

HINT

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

Source

JSOI第二轮Contest1

Analysis

即使对这道题用 Trie 的复杂度令人犹豫，，，= =好像还是不少人写的 Trie

以下代码实现了一个可以指定已知前缀进行查询的 Trie

=w= 不知道为什么博客园复制 Codeforces 的题目会乱版

相比BZOJ就不会了：虽然用户体验不好但是起码版式还算正常

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#define maxn 202000
using namespace std;

char s[maxn];
int trie[maxn][30],CNT;
int end[maxn],tclock = 1,n,m;

void insert(char *str){
    int len = strlen(str),pos = 0;
    for(int i = 0;i < len;i++){
        if(!trie[pos][str[i]-'a']) trie[pos][str[i]-'a'] = ++CNT;
        pos = trie[pos][str[i]-'a'];
    }end[pos] = 1;
}

int find(char *str,int k = 0,int pos = 0){
    int len = strlen(str);
    for(int i = 0;i < len;i++){
        if(!trie[pos][str[i]-'a']) return 0;
        pos = trie[pos][str[i]-'a'];
    }if(k && end[pos]) return 1;
    if(end[pos] && end[pos] != tclock){
        end[pos] = tclock;
//        printf("Get in #%d\n",pos);
        return 1;
    }
    return 0;
}

int main(){
    scanf("%d%d",&n,&m);
    
    while(n--){
        scanf("%s",s);
        insert(s);
    }
    
//    for(int i = 0;i <= CNT;i++){
//        for(int j = 0;j < 26;j++){
//            printf("%d ",trie[i][j]);
//        }cout << endl;
//    }
    
    while(m--){
        scanf("%s",s);
        int len = strlen(s);
        
        if(find(s,1)){
            cout << -1 << endl;
            continue;
        }
        
        tclock++;
        
        // Add
        int pos = 0,ans = 0;
        for(int i = 0;i <= len;i++){
            for(int j = 0;j < 26;j++){
                
                if(!trie[pos][j]) continue;
                int cnt = trie[pos][j];
                if(find(s+i,0,cnt)){
                    ans++;
//                    cout << "pos" << pos << endl;
//                    printf("a%d %d\n",i,j);
//                    cout << s+i << endl;
                }
//                ans += find(s+i,0,cnt);
                
            }
            if(!trie[pos][s[i]-'a']) break;
            pos = trie[pos][s[i]-'a'];
        }
        
        // Delete
        pos = 0;
        for(int i = 0;i < len;i++){
            if(find(s+i+1,0,pos)){
                ans++;
//                printf("-%d\n",i);
            }
            if(!trie[pos][s[i]-'a']) break;
            pos = trie[pos][s[i]-'a'];
        }
        
        // Change
        pos = 0;
        for(int i = 0;i < len;i++){
            for(int j = 0;j < 26;j++){
                if(!trie[pos][j]) continue;
                int cnt = trie[pos][j];
                if(find(s+i+1,0,cnt)){
                    ans++;
//                    printf("@%d %d\n",i,j);
                }
            }
            if(!trie[pos][s[i]-'a']) break;
            pos = trie[pos][s[i]-'a'];
        }
        
        printf("%d\n",ans);
    }
    
    return 0;
}