jfz-2023-s474155/TaskB00/run.py

import sys

# global fsa_description
# global inputWords
# global outputResult


fsa_description = sys.argv[1]
inputWords = sys.argv[2]
outputResult = sys.argv[3]

# fsa_description = "fsa_description.arg"
# inputWords = "test1.in"
# outputResult = "test1.out"

def readDescription(currentPos, charInput):
    with open(fsa_description, 'r', encoding="utf-8") as descript:
        for line in descript:
            if len(line.strip().split('\t')) == 1:
                return line[0], 0
            transition = line.strip().split('\t')
            if len(transition) == 3 and currentPos == transition[0] and charInput == transition[2]:
                return transition[1], 1
    return None, None

answersYN = ''
with open(inputWords, 'r', encoding="utf-8") as inputString:
    for line in inputString:
        pos = '0'
        for char in line:
            if char == '\n':
                answersYN+=('1' if pos == '3' else '0')
            newPos, flag = readDescription(pos, char)
            if flag == 1:
                pos = newPos

print(answersYN)
with open(outputResult, 'w', encoding="utf-8") as output:
    for i in range(0, len(answersYN)):
        if answersYN[i] == '1':
            output.write("YES\n")
        elif answersYN[i] == '0':
            output.write("NO\n")