jfz-2023-s473579/TaskB00/run.py

import sys
def write_answer(answer):
    with open(output_file, 'a') as file:
        file.write(answer+'\n')
def find_next_position(position, character):
    with open(used_table, 'r') as readed_used_table:
        for row_used_table in readed_used_table:
            line = row_used_table.strip().split('\t')
            if position == line[0] and character == line[2]:
                return True,line[1]

# used_table = 'fsa_description.arg'
# input_file = 'test1.in'
# output_file = 'test1.out'
used_table = sys.argv[1]
input_file = sys.argv[2]
output_file = sys.argv[3]
with open(output_file, 'w') as readed_output_file:
    with open(input_file, 'r') as readed_input_file:
        for row_input_file in readed_input_file:
            result  = False
            next_position = None
            position = '0'
            for character in row_input_file:
                if character =='\n':
                    if position=='3':
                        write_answer('YES')
                        break
                    else:
                        write_answer('NO')
                        break
                result, next_position = find_next_position(position,character)
                if result == True:
                    position = next_position