djfz-2021/TaskB00/run.py

80 lines
1.7 KiB
Python
Raw Normal View History

2021-10-23 18:29:55 +02:00
#!/usr/bin/python
import sys
class FSA():
def __init__(self,):
self.initial_state = '0'
self.final_states = set()
self.transitions = dict()
self.alphabet = set()
def add_transition(self, state_from, state_to, symbol):
if state_from in self.transitions.keys():
self.transitions[state_from][symbol] = state_to
else:
self.transitions[state_from] = dict()
self.transitions[state_from][symbol] = state_to
def add_final_state(self, state):
self.final_states.add(state)
def get_final_state(self, string):
current_state = self.initial_state
for symbol in string:
current_state = self.transitions[current_state][symbol]
return current_state
def accepts(self,string):
if self.get_final_state(string) in self.final_states:
return True
else:
return False
fsa = FSA()
# fsa.add_transition(0, 1, 'a')
# fsa.add_transition(1, 2, 'b')
# fsa.add_transition(2, 3, 'c')
# fsa.add_final_state(3)
# fsa.add_final_state(2)
#
# print(fsa.accepts('abc'))
# print(fsa.accepts('ab'))
# print(fsa.accepts('abd'))
table = open(sys.argv[1])
for line in table:
line = line.rstrip('\n')
if len(line.split('\t')) == 3:
a, b, c = line.split('\t')
fsa.add_transition(a, b, c)
fsa.alphabet.add(c)
elif len(line.split('\t')) == 1:
fsa.add_final_state(line)
else:
assert False
for line in sys.stdin:
line = line.rstrip()
line_n = list(line)
for i in range(len(line_n)):
if line_n[i] not in fsa.alphabet:
line_n[i] = 'x'
if fsa.accepts(line_n):
print('YES')
else:
print('NO')