jfz-2023-s473579/TaskC06/run.py
2023-11-25 23:47:56 +01:00

54 lines
1.9 KiB
Python

import sys
def write_answer(answer, output_file):
with open(output_file, 'a', encoding='utf-8') as file:
file.write(answer+'\n')
def run_automaton(file_path, end_positions, output_file):
transitions = {}
with open(file_path, 'r', encoding='utf-8') as file:
with open(output_file, 'w', encoding='utf-8') as file_writed:
lines = file.readlines()
for line in lines:
if line.strip(): # Пропускаем пустые строки
parts = line.split()
state_from = int(parts[0])
if len(parts) > 1:
state_to = int(parts[1])
if len(parts) == 3:
symbol = parts[2]
transitions.setdefault(state_from, []).append((symbol, state_to))
else:
transitions[state_from] = []
def explore_paths(current_state, current_word):
if current_state not in transitions:
if str(current_state) in end_positions:
write_answer(current_word, output_file)
return
possible_transitions = transitions[current_state]
for transition_symbol, next_state in possible_transitions:
explore_paths(next_state, current_word + transition_symbol)
explore_paths(0, "")
def what_is_the_ended_positions(file_path):
array = []
with open(file_path, 'r', encoding='utf-8') as file:
for row in file:
line = row.strip().split('\t')
if len(line) == 1:
array += line
return array
input_file = sys.argv[1]
output_file = sys.argv[2]
# file_path = "small.in"
# output_file = "small.out"
end_positions = what_is_the_ended_positions(input_file)
run_automaton(input_file, end_positions, output_file)