jfz-2023-s473616/TaskB00/run.py
Pawel Felcyn a09908bb53 b00
2023-11-11 19:28:17 +01:00

36 lines
1.2 KiB
Python

#!/usr/bin/env python3
import pandas as pd
import os
import sys
def automat_fun(recursive_df: pd.DataFrame):
first_row = recursive_df.iloc[0]
recursive_df = recursive_df.iloc[1:,:]
return lambda char, state: first_row[1] if state == first_row[0] and char == first_row[2] else automat_fun(recursive_df)(char, state)
def create_automat_from_file(file_path: str):
df = pd.read_csv(file_path, sep='\t', header=None)
accept_state = df.tail(1).iloc[0, 0]
df = df.iloc[:-1,:]
return accept_state, automat_fun(df)
file_path = os.path.join(os.path.dirname(__file__), 'fsa_description.arg')
accept_state, automat_function = create_automat_from_file(file_path)
input_file_path = os.path.join(os.path.dirname(__file__), 'test1.in')
output_file_path = os.path.join(os.path.dirname(__file__), 'test1.out')
results: list[str] = []
with open(input_file_path, 'r', encoding='utf8') as input_file:
for line in input_file:
state = 0
for c in line:
if c == '\n':
continue
state = automat_function(c, state)
results.append('YES\n' if state == accept_state else 'NO\n')
with open(output_file_path, 'w') as output_file:
output_file.writelines(results)