aoc2023/d12/part1.len.py

38 lines
921 B
Python

import sys
def valid(pat:str, nums:list[int]):
pat += '.'
c = 0
for i,p in enumerate(pat):
if p == '#':
c += 1
else:
if c>0:
if c == nums[0]:
c = 0
nums = nums[1:]
if len(nums) == 0:
return 1 if pat[i:].find('#')<0 else 0
else:
return 0
return 0
def next(pattern, nums):
i = pattern.find('?')
if i < 0:
v = valid(pattern, nums)
if v == 1:
print("valid", pattern)
return v
else:
return next(pattern[:i]+'.'+pattern[i+1:], nums)+next(pattern[:i]+'#'+pattern[i+1:], nums)
S= 0
for i,l in enumerate(sys.stdin.read().splitlines()):
pattern, snums = l.split(' ')
nums = list(map(int,snums.split(',')))
n = next(pattern, nums)
print(i,l,n)
S += n
print(S)