POJ1035_Spell checker_KEY

题目传送门

一道暴力可以过的水题。（直接暴力模拟的那种）

但是我打Trie练练模板，但是TMD因为数组开太小卡了好久。

code：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <map>
#include <algorithm>
using namespace std;

struct trie{
    int tr[][],cnt,v[];
    trie(){memset(tr,,sizeof tr),memset(v,,sizeof v),cnt=;}
    void add(string a)
    {
        int now=,len=a.size();
            for(int i=;i<len;i++){
                if(!tr[now][a[i]-'a'])tr[now][a[i]-'a']=++cnt,now=cnt;
                else now=tr[now][a[i]-'a'];
            }
        v[now]=;
    }

    int query(string a)
    {
         int now=,len=a.size(),o=;
             for(int i=;i<len;i++){
                 if(!tr[now][a[i]-'a'])return ;
                 now=tr[now][a[i]-'a'];
             }
        return v[now];
    }
}M;

map<string,int>MP;
struct node{
    string f;int id;
    node(){f="";id=;}
}as[];
string S,ks,t;
int tot=;
inline int cmp(node x,node y){return x.id<y.id;}
int main()
{
//    freopen("x.txt","r",stdin);
    t="";
    int ide=;
    while(cin>>S,S[]!='#'){if(MP[S])continue;M.add(S);MP[S]=++ide;}
    while(cin>>S,S[]!='#'){
        if(M.query(S)){
            cout<<S;cout<<" is correct";puts("");
        }
        else{
            tot=;
            cout<<S<<':';
            int len=S.size();
            for(int i=;i<len;i++){
                for(int j=;j<;j++){
                    t=S[i];
                    ks=j+'a';
                    S.erase(i,),S.insert(i,ks);
                    if(M.query(S))
                        as[++tot].f=S,as[tot].id=MP[S];
                    S.erase(i,),S.insert(i,t);
                }
            }
            for(int i=;i<=len;i++){
                for(int j=;j<;j++){
                    ks=j+'a';
                    S.insert(i,ks);
                    if(M.query(S))
                        as[++tot].f=S,as[tot].id=MP[S];
                    S.erase(i,);
                }
            }
            for(int i=;i<len;i++){
                t=S[i];
                S.erase(i,);
                if(M.query(S))
                    as[++tot].f=S,as[tot].id=MP[S];
                S.insert(i,t);
            }
            sort(as+,as+tot+,cmp);
            for(int i=;i<=tot;i++)
                if(as[i].f!=as[i-].f)
                    cout<<' '<<as[i].f;
            puts("");
        }
    }
}