Datei verschieben

1 files changed, 26 insertions, 18 deletions

diff --git a/alphametiken.py b/alphametiken.py
index 2091627..c98f7aa 100755
--- a/alphametiken.py
+++ b/alphametiken.py
@@ -4,20 +4,27 @@
 
 import re
 
-Alphametik = "SUCHEN - MACHT = SPASS"
-
-# Liste der Buchstaben erstellen
-buchstaben = []
-for c in Alphametik:
-    if (c not in "+-*/= ") and (c not in buchstaben):
-        buchstaben.append(c)
+Alphametik = "SUCHEN-MACHT=SPASS"
 
 lösungen = []
 
+# Liste der Buchstaben erstellen aus `wort`
+def buchstaben_liste(wort):
+    buchstaben = []
+    for c in wort:
+        if (c not in "+-*/= ") and (c not in buchstaben):
+            buchstaben.append(c)
+
+    return buchstaben
+
 # Ersetzt einen Buchstaben aus `b` in einem Wort `wort` nacheinander durch `0-9`
 # und ruft sich selbst auf. Ist `b` leer, wird das Alphametik auf Richtigkeit getestet
 # und ggf. zu `lösungen` hinzugefügt.
-def ersetze(b, wort, benutzte_zahlen = []):
+# Zurückgegeben wird eine Liste mit Lösungen, Element der Liste ist der Rechnungsstring und ein dict
+def ersetze(b, wort, benutzte_zahlen = [], lösungen = [], alle_buchstaben = None):
+    if alle_buchstaben == None:
+        alle_buchstaben = b
+
     if len(b):
         for j in range(10):
             if j in benutzte_zahlen: # jede Zahl nur einmal verwenden
@@ -27,24 +34,25 @@ def ersetze(b, wort, benutzte_zahlen = []):
             m_benutzte_zahlen.append(j)
 
             m_wort = wort.replace(b[0], str(j))
-            ersetze(b[1:], m_wort, m_benutzte_zahlen)
+            ersetze(b[1:], m_wort, m_benutzte_zahlen, lösungen, alle_buchstaben)
     else:
         if re.search(r"\b0", wort): # 0 als erste Ziffer in einer Zahl ist nicht erlaubt
             return
 
-        if eval(wort):
-            lsg = dict()
-            for j, k in zip(buchstaben, benutzte_zahlen):
-                lsg[j] = k
+        if eval(wort.replace("=", "==")):
+            lsg = [wort, dict()]
+            for j, k in zip(alle_buchstaben, benutzte_zahlen):
+                lsg[1][j] = k
 
-            print(lsg) # DEBUG
             lösungen.append(lsg)
 
-w = Alphametik.replace("=", "==")
-ersetze(buchstaben, w)
+    return lösungen
+
+
+l = ersetze(buchstaben_liste(Alphametik), Alphametik)
 
-if len(valides) == 0:
+if len(l) == 0:
     print("Keine Lösungen vorhanden")
 else:
     print("Es konnten folgende Lösungen gefunden werden:")
-    print(lösungen)
+    print(l)