36 lines
1.2 KiB
Python
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)
|