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

来源：互联网发布：知乎女神王珏编辑：程序博客网时间：2024/05/21 15:50

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;}

0 0