Add task197 solver (267/267)

medal-solvers/task_solvers/task197_solver_267.py

@@ -0,0 +1,84 @@
+"""
+Task 197 Python Solver: Pattern repetition with color substitution.
+267/267 verified.
+
+Rule:
+1. One row is the "template" (fully non-zero across entire width)
+2. Other non-zero rows are "seeds" (first N pixels filled, rest zero)
+3. For each seed row: build color mapping from template → seed 
+   (template[c] maps to seed[c] for c < seed_length)
+4. Apply mapping to full template to fill the entire row
+"""
+import numpy as np
+import json
+
+
+def solve_task197(grid):
+    grid = np.array(grid)
+    H, W = grid.shape
+    out = grid.copy()
+    
+    # Find template row (fully non-zero)
+    template_row = None
+    for r in range(H):
+        if np.all(grid[r] != 0):
+            template_row = r
+            break
+    if template_row is None:
+        return out.tolist()
+    
+    template = grid[template_row]
+    
+    # For each seed row
+    for r in range(H):
+        if r == template_row:
+            continue
+        row = grid[r]
+        if np.all(row == 0):
+            continue
+        
+        # Find seed length (contiguous non-zero from left)
+        seed_len = 0
+        for c in range(W):
+            if row[c] != 0:
+                seed_len = c + 1
+            else:
+                break
+        if seed_len == 0:
+            continue
+        
+        # Build color mapping: template_color → seed_color
+        color_map = {}
+        for c in range(seed_len):
+            t_color = template[c]
+            s_color = row[c]
+            if t_color not in color_map:
+                color_map[t_color] = s_color
+        
+        # Apply mapping to full template
+        for c in range(W):
+            t_color = template[c]
+            if t_color in color_map:
+                out[r, c] = color_map[t_color]
+    
+    return out.tolist()
+
+
+def validate():
+    with open('/kaggle/input/competitions/neurogolf-2026/task197.json') as f:
+        data = json.load(f)
+    
+    all_ex = data.get('train', []) + data.get('test', []) + data.get('arc-gen', [])
+    right, wrong = 0, 0
+    for ex in all_ex:
+        result = solve_task197(ex['input'])
+        if result == ex['output']:
+            right += 1
+        else:
+            wrong += 1
+    print(f"Task 197: {right}/{right+wrong}")
+    return wrong == 0
+
+
+if __name__ == '__main__':
+    validate()