【codeforces 514C】Watto and Mechanism(字典树做法)

【题目链接】:http://codeforces.com/contest/514/problem/C

【题意】



给你n个字符串;

然后给你m个询问;->m个字符串

对于每一个询问字符串

你需要在n个字符串里面找到和它的长度相同,且只有一个位置的字符不同的字符串;

或者告知这是不存在的;

【题解】



写个字典树;

在每一位多了两种选择;

(即更改这个字符)

然后前面已经确定匹配的不要重新匹配;->不然会超时

有改和不改两种可能.

注意最后必须要改一个字符.

字典树以后还是用0作为根节点吧。不然tot老是忘记赋初值1;



【Number Of WA】



4



【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define ps push_back
#define fi first
#define se second
#define rei(x) scanf("%d",&x)
#define rel(x) scanf("%lld",&x)
#define ref(x) scanf("%lf",&x)
#define ms(x,y) memset(x,y,sizeof x)

typedef pair<int,int> pii;
typedef pair<LL,LL> pll;

const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);
const int N = 1e6+1000;

struct node
{
    int c[3],flag;
};

int n,m,tot=1,len;
node tree[N];
string s;
//fi(1,0,0)
bool fi(int temp,int pos,int gai)
{
    if (pos>=len)
        return gai&&tree[temp].flag;
    if (!gai)
        for (char a='a';a<='c';a++)
            if (a!=s[pos])
            {
                if (tree[temp].c[a-'a'] && fi(tree[temp].c[a-'a'],pos+1,1))
                    return true;
            }
    if (tree[temp].c[s[pos]-'a'])
        return fi(tree[temp].c[s[pos]-'a'],pos+1,gai);
    return false;
}

int main()
{
    //freopen("F:\\rush.txt","r",stdin);
    rei(n),rei(m);
    rep1(i,1,n)
    {
        cin >> s;
        len = s.size();
        int temp = 1;
        rep1(j,0,len-1)
        {
            if (!tree[temp].c[s[j]-'a'])
                tree[temp].c[s[j]-'a']=++tot;
            temp = tree[temp].c[s[j]-'a'];
        }
        tree[temp].flag = 1;
    }
    rep1(i,1,m)
    {
        cin >> s;
        len = s.size();
        if (fi(1,0,0))
            puts("YES");
        else
            puts("NO");
    }
    //printf("\n%.2lf sec \n", (double)clock() / CLOCKS_PER_SEC);
    return 0;
}