From 33eadbdbb1c143bb8be9a959440dd938340ce8e6 Mon Sep 17 00:00:00 2001
From: Daniel Weber <daniel.weber@gentex.com>
Date: Thu, 19 Dec 2024 11:37:27 -0500
Subject: [PATCH] p2

---
 2024/day19/main.py | 52 +++++++++++++++-------------------------------
 1 file changed, 17 insertions(+), 35 deletions(-)

diff --git a/2024/day19/main.py b/2024/day19/main.py
index 47f0ee2..cc52e19 100644
--- a/2024/day19/main.py
+++ b/2024/day19/main.py
@@ -14,33 +14,24 @@ def check_matchp1(p, t):
                     return True
     return False
 
+G = {}
 def match_count(p, t):
     count = 0
-    if p=='':
-        return count
-    for ts in t:
-        if p == ts:
-            count += 1
-        else:
-            if p[:len(ts)] == ts[:len(p)]:
-                sub_count = match_count(p[len(ts):], t)
-                if sub_count > 0:
+    if p in G:
+        return G[p]
+    else:
+        if p=='':
+            return count
+        for ts in t:
+            if p == ts:
+                count += 1
+            else:
+                if p[:len(ts)] == ts[:len(p)]:
+                    sub_count = match_count(p[len(ts):], t)
                     count += sub_count
+        G[p] = count
     return count
 
-def find_substrings(p, t):
-    ss = []
-    if p=='':
-        return ss
-    for ts in t:
-        if p == ts:
-            ss.append(ts)
-        else:
-            if p[:len(ts)] == ts[:len(p)]:
-                ss.append(find_substrings(p[len(ts):], t))
-    return ss
-
-
 def part_one(p, t):
     p1=0
     for ps in p:
@@ -49,25 +40,16 @@ def part_one(p, t):
     print("p1: ", p1)
 
 def part_two(p, t):
-    p2=0
-    tset = set(t)
-    for ts in tset:
-        print("SUBSTRING: ", ts)
-        print(find_substrings(ts, tset))
-
-    # for c, ps in enumerate(p):
-    #     p2 += match_count(ps, t)
-
+    p2 = 0
+    for ps in p:
+        p2 += match_count(ps, t)
     print("p2: ", p2)
 
 def main():
-    f = open('test.txt', 'r')
+    f = open('input.txt', 'r')
     contents =  [x.strip() for x in f.readlines()]
-    print(contents)
     towels = contents[0].split(", ")
     patterns = contents[2:]
-    print(patterns)
-    print(towels)
     part_one(patterns, towels)
     part_two(patterns, towels)