6-02

06/06-2.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# file = "./ex.txt"
+file = "./input.txt"
+
+def max_length(li:list[str]) -> int:
+    res = 0
+    for l in li:
+        if len(l) > res:
+            res = len(l)
+    return res
+
+def expand_lines(li:list[str], nl:int) -> list[str]:
+    res = []
+    for l in li:
+        while len(l) < nl:
+            l += "x"
+        res.append(l)
+    return res
+
+def get_numbers(li:list[str]) -> list[int]:
+    res = []
+    part = []
+    num = ""
+    for pos in range(len(li[0])):
+        for l in li:
+            if l[pos] != "x":
+                num += l[pos]
+        if num != "":
+            part.append(int(num))
+            num = ""
+        else:
+            res.append(part)
+            part = []
+    res.append(part)
+    return res
+
+
+if __name__ == "__main__":
+    solution = 0
+    lines = []
+    operator_list = []
+    input_file = open(file, "r")
+    for line in input_file:
+        line = line.replace(" ","x")
+        line = line.strip()
+        if line == "":
+            continue
+        elif "*" in line or "+" in line:
+            operator_list = line.replace("x"," ").split()
+        else:
+            lines.append(line)
+
+    lines = expand_lines(lines, max_length(lines))
+    num_list = get_numbers(lines)
+    for i in range(len(operator_list)):
+        if operator_list[i] == "+":
+            op_result = 0
+            for n in num_list[i]:
+                op_result += n
+        elif operator_list[i] == "*":
+            op_result = 1
+            for n in num_list[i]:
+                op_result *= n
+        else:
+            op_result = None
+        solution += op_result
+
+    print(f"Solution: {solution}")