【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("
%.2lf sec 
", (double)clock() / CLOCKS_PER_SEC);
    return 0;
}