From 216b5843f184e2cd4353c79f6134291019697e77 Mon Sep 17 00:00:00 2001
From: Eryk Miszczuk <erymis@st.amu.edu.pl>
Date: Mon, 2 Dec 2019 21:39:01 +0100
Subject: [PATCH] Untested TaskB02

---
 TaskB02/run    |  2 ++
 TaskB02/run.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+)
 create mode 100644 TaskB02/run
 create mode 100644 TaskB02/run.py

diff --git a/TaskB02/run b/TaskB02/run
new file mode 100644
index 0000000..c7ddeed
--- /dev/null
+++ b/TaskB02/run
@@ -0,0 +1,2 @@
+#!/bin/bash
+python TaskB02/run.py "$@"
diff --git a/TaskB02/run.py b/TaskB02/run.py
new file mode 100644
index 0000000..1072c90
--- /dev/null
+++ b/TaskB02/run.py
@@ -0,0 +1,45 @@
+import re
+import sys
+
+atandt_desc_pattern = re.compile(r"^([0-9]+) ([0-9]+) (\S+)( [0-9]+\.[0-9]+)?$")
+atandt_accepted_state_pattern = re.compile(r"^[0-9]+$")
+
+current_state = 0
+accept_states = []
+
+state_move = {}
+
+for line in sys.stdin:
+    line = line.replace('\n', '')
+    move = atandt_desc_pattern.match(line)
+    succes = atandt_accepted_state_pattern.match(line)
+    if move:
+        if move.group(4):
+            state_move[(int(move.group(1)), move.group(3))] = (int(move.group(2)), float(move.group(4)))
+        else:
+            state_move[(int(move.group(1)), move.group(3))] = (int(move.group(2)), 0)
+    if succes:
+        accept_states.append(int(line))
+
+with open(sys.argv[1], 'r', encoding='utf-8') as f:
+    for word in f:
+        current_state = 0
+        word = word.replace('\n', '')
+        character_number = 0
+        while character_number < len(word):
+            if ((current_state, word[character_number]) in state_move):
+                current_state = state_move[(current_state, word[character_number])][1]
+                character_number += 1
+            elif (current_state, '<eps>') in state_move):
+                current_state = state_move[(current_state, '<eps>')][0]
+            else:
+                current_state = -1
+                character_number += 1
+        
+        while ((current_state, '<eps>') in state_move):
+            current_state = state_move[(current_state, '<eps>')][0]
+            
+        if (current_state in accept_states):
+            print("YES " + word)
+        else:
+            print("NO" + word)