Codeforces Round #291 (Div. 2) C - Watto and Mechanism 字符串

【题意】给n个字符串组成的集合，然后有m个询问(0 ≤ n ≤ 3·10⁵, 0 ≤ m ≤ 3·10⁵) ，每个询问都给出一个字符串s，问集合中是否存在一个字符串t，使得s和t长度相同，并且仅有一个字符不同。(字符串总长度为6·10⁵)，所有字符只有a,b,c。

【题解】因为只有三种字符，用Trie最合适。先把n个字符串插入到Trie中。然后每读入一个字符串，首先枚举差异字符的位置，依次替换为另外两种字符，并检查是否合法。如果合法就直接输出YES。

显然每替换一个字符就从头检查一遍太浪费时间，所以保存字典树中差异字符位置的指针，每次从该位置开始检查即可。

我一开始是直接处理n个字符串，枚举差异字符的位置，分别替换后加入到字典树中。但是这样做会耗费相当多的空间，空间复杂度是指数级别(3^len)。MLE或者RE。但是一直返回的是WA7，QAQ找了半天错。。

#include<bits/stdc++.h>
#define eps 1e-9
#define FOR(i,j,k) for(int i=j;i<=k;i++)
#define MAXN 1000005
#define MAXM 40005
#define INF 0x3fffffff
using namespace std;
typedef long long LL;
int i,j,k,n,m,x,y,T,ans,big,cas,num,len;
bool flag;
char s[1000005];

struct Trie
{
    int ch[1000005][26];
    int val[1000005];
    int size;
    Trie()
    {
        size=1;
        memset(ch[0],0,sizeof(ch[0]));
        val[0]=0;
        val[1]=0;
    }

    int idx(char c)
    {
        return c-‘a‘;
    }

    void insert(char *s,int v)
    {
        int u=0,len=strlen(s);
        for (int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if (!ch[u][c])
            {
                memset(ch[size],0,sizeof(ch[size]));
                val[size]=0;
                ch[u][c]=size++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }

    bool exist(char *s)
    {
        int u=0,len=strlen(s);
        for (int i=0;i<len;i++)
        {
            int c=idx(s[i]);

            for (int j=0;j<3;j++)
            {
                if (c==j || ch[u][j]==0) continue;
                int u2=ch[u][j];
                flag=true;
                for (int k=i+1;k<len;k++)
                {
                    int c=idx(s[k]);
                    if (!ch[u2][c])
                    {
                        flag=false;
                        break;
                    }
                    u2=ch[u2][c];
                }
                if (flag && val[u2])
                {
                    return true;
                }
            }

            if (!ch[u][c])
            {
                return false;
            }
            u=ch[u][c];
        }
        return false;
    }
} tree;

int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
    {
        scanf("%s",s);
        tree.insert(s,1);
    }
    for (i=1;i<=m;i++)
    {
        scanf("%s",s);
        if (tree.exist(s)) puts("YES");
        else puts("NO");
    }

    return 0;
}