djfz-2020-s444399/TaskE00/run

28 lines
636 B
Plaintext
Raw Normal View History

2020-12-10 14:34:21 +01:00
#!/usr/bin/python3
import sys
import re
sciezki = {};
stanyakceptujace = [];
f = open(sys.argv[1], "r");
lines = f.readlines();
for line in lines:
answer = re.match('^([0-9]+) ([0-9]+) ([a-z]*)$', line);
if answer:
try:
sciezki[answer.group(1)][answer.group(3)] = answer.group(2);
except KeyError:
sciezki.update({answer.group(1):{}})
sciezki[answer.group(1)][answer.group(3)] = answer.group(2);
else:
stanyakceptujace.append(int(line[:-1]));
for input in sys.stdin.readlines():
act = 0;
for char in input[:-1]:
act = int(sciezki[str(act)][char])
if act in stanyakceptujace:
print("YES")
else:
print("NO")