Watto and Mechanism

考虑字典树上深搜。

由于字符集只有 abc\texttt{abc},所以深搜是可过的。

建立完字典树后,从根开始,维护现在是否有存在一个不同的字符,如果有,那么下一个字符肯定和询问的串相同,如果没有不同,那么下一个可以选择相同也可以选择不同的。

#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;
}
posted @   HappyBobb  阅读(1)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示