Watto and Mechanism
考虑字典树上深搜。
由于字符集只有 ,所以深搜是可过的。
建立完字典树后,从根开始,维护现在是否有存在一个不同的字符,如果有,那么下一个字符肯定和询问的串相同,如果没有不同,那么下一个可以选择相同也可以选择不同的。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
using namespace std;
const int N = 1e6 + 5;
string s, p;
int n, m;
class Trie
{
public:
int son[N][4], idx, rt;
void insert(string x)
{
int u = rt;
for (auto c : x)
{
int p = c - 'a';
if (!son[u][p]) son[u][p] = ++idx;
u = son[u][p];
}
}
bool dfs(int x, int u, bool g)
{
if (!u && x != 0) return 0;
if (x == p.size() && g) return 1;
if (g)
{
return dfs(x + 1, son[u][p[x] - 'a'], g);
}
else
{
for (int j = 0; j <= 2; j++)
{
if (p[x] - 'a' == j)
{
if (dfs(x + 1, son[u][j], 0)) return 1;
}
else
{
if (dfs(x + 1, son[u][j], 1)) return 1;
}
}
return 0;
}
}
}t;
int main()
{
ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
cin >> s;
t.insert(s);
}
while (m--)
{
cin >> p;
cout << (t.dfs(0, t.rt, 0) ? "YES" : "NO") << "\n";
}
return 0;
}

浙公网安备 33010602011771号