leichte Verbesserungen an der Mobile-Lösung

1 files changed, 10 insertions, 6 deletions

diff --git a/mobile.py b/mobile.py
index a5c5bdd..86c0fdd 100755
--- a/mobile.py
+++ b/mobile.py
@@ -3,9 +3,9 @@ from math import floor
 import random
 
 # python list mit Elementen in der Form (Position, [Gewicht oder Liste]), wobei Position auf der linken Seite negativ ist
-mobile = [(-4, [(-2, [(0, 1)]), (1, [(0, 2)])]), (-1, [(0, 2)]), (2, [(0, 3)]), (4, [(0, 2)])]
+Mobile = [(-4, [(-2, [(0, 1)]), (1, [(0, 2)])]), (-1, [(0, 2)]), (2, [(0, 3)]), (4, [(0, 2)])]
 
-figuren = [2, 3, 1, 2, 3]
+Figuren = [2, 3, 1, 2, 3]
 
 # gibt eine Liste zurück, in der die Elemente aus `fig` auf 4er-Balken aufgeteilt sind
 def split(fig):
@@ -56,7 +56,11 @@ def weight(fig):
 
     return res
 
-r = list([(j, j) for j in figuren]) # der Input wird aufbereitet
-f = split(r)
-w = weight(f)
-print(w)
+def mobile(liste):
+    r = list([(j, j) for j in liste]) # der Input wird aufbereitet
+    f = split(r)
+    w = weight(f)
+    return w
+
+if __name__ == "__main__":
+    print(mobile(Figuren))