diff --git a/TaskC00/run.py b/TaskC00/run.py index 537c832..14d12a4 100644 --- a/TaskC00/run.py +++ b/TaskC00/run.py @@ -17,11 +17,13 @@ class Automat: return random.choice(end_states) def consume_word(self, word: str): + prev_state = 0 state = 0 for char in word: + prev_state = state state = self._consume_character(state, char) if state is None: - return False + return prev_state in self._end_states return state in self._end_states def create_automat_from_file(file_path: str): diff --git a/TaskC01/run.py b/TaskC01/run.py new file mode 100644 index 0000000..68ae170 --- /dev/null +++ b/TaskC01/run.py @@ -0,0 +1,55 @@ +import pandas as pd +import os +import sys +import random + +class Automat: + def __init__(self, df: pd.DataFrame, end_states: list): + self._df = df + self._end_states = end_states + + def _consume_character(self, state: int, char: str): + filter = (self._df[0] == state) & (self._df[2] == char) + matching_rows = self._df[filter] + if matching_rows.empty: + return None + end_states = matching_rows.iloc[:,1].tolist() + return random.choice(end_states) + + def consume_word(self, word: str): + prev_state = 0 + state = 0 + for char in word: + prev_state = state + state = self._consume_character(state, char) + if state is None: + return prev_state in self._end_states + return state in self._end_states + +def create_automat_from_file(file_path: str): + automat_df = pd.read_csv(file_path, sep=' ', header=None) + end_states_condition = (automat_df[1].isnull()) & (automat_df[2].isnull()) + end_states_df = automat_df[end_states_condition] + end_states = end_states_df.iloc[:,0].tolist() + + automat_df = automat_df[~end_states_condition] + automat_df[2] = automat_df[2].astype(str) + + return Automat(automat_df, end_states) + +automat_path = os.path.join(os.path.dirname(__file__), 'test.arg') +automat = create_automat_from_file(automat_path) + +test_path = os.path.join(os.path.dirname(__file__), 'test.in') + + +results = [] +with open(test_path, 'r') as f: + for line in f: + results.append('YES' if automat.consume_word(line.strip()) else 'NO') + +out_path = os.path.join(os.path.dirname(__file__), 'test.out') + +with open(out_path, 'w') as f: + for result in results: + f.write(result + '\n') \ No newline at end of file diff --git a/TaskC01/test.arg b/TaskC01/test.arg new file mode 100644 index 0000000..afd591a --- /dev/null +++ b/TaskC01/test.arg @@ -0,0 +1,5 @@ +0 1 a +0 1 b +0 1 c +1 2 b +2 diff --git a/TaskC01/test.out b/TaskC01/test.out new file mode 100644 index 0000000..f47b8ff --- /dev/null +++ b/TaskC01/test.out @@ -0,0 +1,6 @@ +YES +YES +NO +NO +YES +NO