Codeforces Round #291 (Div. 2)C.Watto and Mechanism——字典树+dfs

时间:2022-12-30 13:22:05

http://codeforces.com/contest/514

n个字符串m个询问,判断询问的字符串能否通过修改其中的一个字符使得变成给定的n个字符串中的某一个

注意:必须修改一个字符

#include<bits/stdc++.h>
const int maxnode=6e5+100;
const int sigma_size=3;
using namespace std;
int n,m;
int ch[maxnode][sigma_size],val[maxnode];
int sz;
void init(){
    memset(ch,0,sizeof ch);
    memset(val,0,sizeof(val));
    sz=1;
}
void insert(char *s)
{
    int u=0,l=strlen(s);
    for(int i=0; i<l; ++i) {
        int c=s[i]-'a';
        if(!ch[u][c]) {
            ch[u][c]=sz++;
        }
        u=ch[u][c];
    }
    val[u]=1;
}
char s[maxnode];
bool dfs(char *s,int rt,int p,int d,int l)
{
    if(d>1) return false;
    if(p==l){
        if(d==1&&val[rt]) return true;
        return false;
    }
    for(int i=0;i<sigma_size;++i){
        if(ch[rt][i]){
            if(dfs(s,ch[rt][i],p+1,d+(i!=s[p]-'a'),l)) return true;
        }
    }
    return false;
}

int main()
{
    init();
    scanf("%d%d",&n,&m);
    for(int i=0; i<n; ++i) {
        scanf("%s",s);
        insert(s);
    }
    for(int i=0; i<m; ++i) {
        scanf("%s",s);
        if(dfs(s,0,0,0,strlen(s))) printf("YES\n");
        else printf("NO\n");
    }

    return 0;
}