字典树（+DFS）——Watto and Mechanism ( CodeForces 514C )

• 题目链接：
  http://codeforces.com/contest/514/problem/C

• 分析：
  题目是先给出n个字符串和m个字符串，对于m个字符串中每一个字符串都可以在n个字符串中找到一个与之匹配（允许由一位字符的差异，但不允许超出或少于）。一看题目，就能想到是用n 个字符串建树，然后再搜索。

• 题解：
  比较难写的是这个DFS函数，先讨论字符串有没有到底，到底后无错误并且之后没有节点了，就返回true，否则返回false；
  然后讨论下一个字符对应的节点有没有值，有则递归判断。
  最后，如果再某一个递归处发现一个错误，那么继续递归判断，然后返回结果。

• DFS函数：

bool dfs(Trie *tr,int len,int f)
{
        if(s[len]=='\0' )//字符串到底了
        {
            if(f==0 && tr->index==1)  //如果无错误且之后无节点
                return true;
            else
                return false;
        }
        if(tr->next[s[len]-'a']!=0)//字符串未到底，并且字典树之后还有节点
        {
            if(dfs(tr->next[s[len]-'a'],len+1,f))//递归调用
                return true;
        }
        if(f==1)//如果字符串未到底，但是字典树到底了或者不匹配并且诶错误为1，就递归调用
        {
            int i;
            for(i=0;i<=2;i++)
                if(i!=s[len]-'a' && tr->next[i]!=0)
                    if(dfs(tr->next[i],len+1,0))
                        return true;
        }
        return false;
}

• 代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
int T,t,n,m;
char s[1000010];
struct Trie
{
        int index;
        Trie *next[3];
        Trie()
        {
            index=-1;
            memset(next,0,sizeof(next));
        }
};
Trie *root=new Trie;
void Trie_Insert(Trie *tr,int len)
{
        if(s[len]!='\0')
        {
            if(tr->next[s[len]-'a']==0)
                tr->next[s[len]-'a']=new Trie;
            Trie_Insert(tr->next[s[len]-'a'],len+1);
        }
        else
            tr->index=1;   //表示这之后没有节点
}
bool dfs(Trie *tr,int len,int f)
{
        if(s[len]=='\0' )
        {
            if(f==0 && tr->index==1)  //如果无错误且之后无节点
                return true;
            else
                return false;
        }
        if(tr->next[s[len]-'a']!=0)
        {
            if(dfs(tr->next[s[len]-'a'],len+1,f))
                return true;
        }
        if(f==1)
        {
            int i;
            for(i=0;i<=2;i++)
                if(i!=s[len]-'a' && tr->next[i]!=0)
                    if(dfs(tr->next[i],len+1,0))
                        return true;
        }
        return false;
}
int main()
{
        int i,j,k;
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
            scanf("%s",s+1);
            Trie_Insert(root,1);
        }
        for(i=1;i<=m;i++)
        {
            scanf("%s",s+1);
            if(dfs(root,1,1))
                printf("YES\n");
            else
                printf("NO\n");
        }
}