djfz-2023-s464933/TaskC00/run.py

import sys
import copy

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():
            if symbol not in self.transitions[state_from].keys():
                self.transitions[state_from][symbol] = {state_to}
            else:
                self.transitions[state_from][symbol] |= {state_to}
        else:
            self.transitions[state_from] = dict()
            self.transitions[state_from][symbol] = {state_to}
    def get_final_state(self, string):

        current_states = {self.initial_state}
        for symbol in string:
            new_current_states = set()
            for current_state in current_states:
                try:
                    new_current_states |= self.transitions[current_state][symbol]
                except:
                    pass
                current_states = copy.deepcopy(new_current_states)
            return current_states

    def add_final_state(self, state):
        self.final_states.add(state)
    def accepts(self, string):
        final_states = self.get_final_states(string)
        for final_state in final_states:
            if final_state in self.final_states:
                return True
        return False
fsa = FSA()

table = open(sys.argv[1])
for line in table:
    line = line.rstrip()

    if len(line.split('\t')) == 3:
        a, b, c = line.split('\t')
        fsa.add_transition(a, b, c)
        fsa.alphabet.add(c)
    else:
        fsa.add_final_state(line)


for line in sys.stdin:
    line = line.rstrip()

    if fsa.accepts(line):
        print('YES')
    else:
        print('NO')