Codeforces Round #291 (Div. 2) C. Watto and Mechanism Trie

链接：

http://codeforces.com/contest/514/problem/C

题意：

字符串匹配，要求刚好有一个字符不一样

题解：

字典树，直接暴力dfs就可以了，开始以为会超时，其实并不会

代码：

int n, m;
int Trie[MAXN][3], v[MAXN];
int sz = 1;
string s;

void insert(string s) {
    int now = 1;
    rep(i, 0, s.length()) {
        int x = s[i] - 'a';
        if (!Trie[now][x]) Trie[now][x] = ++sz;
        now = Trie[now][x];
    }
    v[now] = 1;
}

bool dfs(int now, int len, int fg) {
    if (len == s.length()) {
        if (v[now] && fg) return true;
        return false;
    }
    int x = s[len] - 'a';
    if (Trie[now][x]) if (dfs(Trie[now][x], len + 1, fg)) return true;
    if (!fg) {
        x = (x + 1) % 3;
        if (Trie[now][x]) if (dfs(Trie[now][x], len + 1, 1)) return true;
        x = (x + 1) % 3;
        if (Trie[now][x]) if (dfs(Trie[now][x], len + 1, 1)) return true;
    }
    return false;
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> n >> m;
    while (n--) {
        cin >> s;
        insert(s);
    }
    while (m--) {
        cin >> s;
        if (dfs(1, 0, 0)) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}