TaskA03/solution.py

@@ -1,34 +1,27 @@
 import sys
 
-# '19xx' literally would be accepted as well but we won't tell anyone
+# return if the sentence contains word pies
 def solve(lines):
-    def detect_date(s):
-        date_f = "19XX r."
-        date_l = len(date_f)
-
-        j = 0
-        for i in range(len(s)):
-            c = s[i]
-
-            if j == date_l:
-                return True
-            elif c == date_f[j] \
-                or (j in (2, 3) and c.isdigit()):
-                j += 1
-                continue
-
-            j = 0
-
-        return False
-
-    return filter(detect_date, lines)
-
+    res = []
+    for l in lines:
+        # first one didnt work what about the other dates?
+        i = 0
+        s = l
+        while i != -1:
+            if i+7 < len(s):
+                if s[i:i+2] == "19" and s[i+2].isdigit() and s[i+3].isdigit() and s[i+4:i+7] == " r.":
+                    res.append(l)
+                    break
+            i += 2 # skipping current date
+            s = s[i:]
+            i = s.find("19")
+    return res
 
 if __name__ == "__main__":
     lines = []
     
     fp = sys.argv[1]
-    with open(fp, encoding="utf-8") as f: # following symlinks might not work on Windows
+    with open("../TaskA02/polish_wiki_excerpt.in", encoding="utf-8") as f:
         lines = f.readlines()
     sol = solve(lines)
     with open("./polish_wiki_excerpt.out", 'w', encoding="utf-8") as f:

TaskA04/solution.py

@@ -1,30 +1,25 @@
 import sys
 
-# detect numbers. separate numbers by spaces
 def solve(lines):
     res = []
 
-    j = 0
     for l in lines:
         numbers = []
-        for i in range(len(l)):
-            c = l[i]
-            
-            if j > 0 and not c.isdigit():
-                numbers.append(l[i-j:i])
-            j = j+1 if (c.isdigit() and c.isascii()) else 0
-
+        for w in l.split():
+            number = ''.join(filter(lambda c: c.isdigit(), w))
+            if number:
+                numbers.append(number)
         if numbers:
-            res.append(' '.join(numbers))
-    return '\n'.join(res)
+            res.append(' '.join(numbers) + '\n')
+
+    return res
 
 if __name__ == "__main__":
     lines = []
     
     fp = sys.argv[1]
-    with open(fp, encoding="utf-8") as f:
+    with open("../TaskA02/polish_wiki_excerpt.in", encoding="utf-8") as f:
         lines = f.readlines()
     sol = solve(lines)
     with open("./polish_wiki_excerpt.out", 'w', encoding="utf-8") as f:
         f.writelines(sol)
-        f.write('\n')  # it expects for file to end with newline :)