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

2120 단어 codeforces
이 문제 CF 공식 문제 해설은 해시로 하지만 해시를 움켜쥐고 사활을 부릴 수 없다 = =...그리고 사전나무를 만들어 샅샅이 뒤질 수 있다는 것을 발견했다.이런 방법은 공간과 시간의 복잡도와 관련이 있는데, 직렬이 가장 길기 때문에 6*10^5.그래서 만들어진 사전은 나무의 갈라짐과 깊이가 많지 않을 거야...시간의 복잡도는 결코 나쁘지 않다...
#include <iostream>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <climits>
#include <cstdlib>
#include <cmath>
#include <time.h>
#define maxn 600005
#define maxm 200005
#define eps 1e-7
#define mod 1000000007
#define INF 0x3f3f3f3f
#define PI (acos(-1.0))
#define lowbit(x) (x&(-x))
#define mp make_pair
#define ls o<<1
#define rs o<<1 | 1
#define lson o<<1, L, mid 
#define rson o<<1 | 1, mid+1, R
#define pii pair<int, int>
#pragma comment(linker, "/STACK:16777216")
typedef long long LL;
typedef unsigned long long ULL;
using namespace std;
LL qpow(LL a, LL b){LL res=1,base=a;while(b){if(b%2)res=res*base;base=base*base;b/=2;}return res;}
LL powmod(LL a, LL b){LL res=1,base=a;while(b){if(b%2)res=res*base%mod;base=base*base%mod;b/=2;}return res;}
//head

int next[maxn][3];
int end[maxn];
char s[maxn];
int root = 1, tail, n, m, ok;

void read()
{
	int u;
	tail = 2;
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++) {
		scanf("%s", s);
		u = root;
		for(int j = 0; s[j]; j++) {
			int c = s[j] - 'a';
			if(!next[u][c]) next[u][c] = tail++;
			u = next[u][c];
		}
		end[u] = true;
	}
}

void dfs(int u, int dep, int k)
{
	if(dep == n && k && end[u]) ok = true;
	if(ok) return;
	if(dep > n) return;
	int c = s[dep] - 'a';
	if(next[u][c]) dfs(next[u][c], dep + 1, k);
	if(!k) for(int i = 0; i < 3; i++) if(i != c && next[u][i]) dfs(next[u][i], dep + 1, 1);
}

void work()
{
	while(m--) {
		scanf("%s", s);
		n = strlen(s);
		ok = 0;
		dfs(root, 0, 0);
		if(ok) printf("YES
"); else printf("NO
"); } } int main() { read(); work(); return 0; }

좋은 웹페이지 즐겨찾기