diff options
author | Thomas Voss <thomasvoss@live.com> | 2021-12-02 13:58:09 +0100 |
---|---|---|
committer | Thomas Voss <thomasvoss@live.com> | 2021-12-02 13:58:09 +0100 |
commit | e257eb53359927c537d2e9c31ca8eae7a0f6dfe7 (patch) | |
tree | dd4c4b339e71d4cda3b8fa6140183e764d01d43c /2020/19/puzzle-2.py | |
parent | 8e6440d8fd69caedc11192dca02a6ae28f5248cf (diff) |
Combine parts 1 and 2 into one file
Diffstat (limited to '2020/19/puzzle-2.py')
-rwxr-xr-x | 2020/19/puzzle-2.py | 84 |
1 files changed, 0 insertions, 84 deletions
diff --git a/2020/19/puzzle-2.py b/2020/19/puzzle-2.py deleted file mode 100755 index b0efe15..0000000 --- a/2020/19/puzzle-2.py +++ /dev/null @@ -1,84 +0,0 @@ -#!/usr/bin/env python3 - -from collections import OrderedDict -from re import search - - -def check(test: str, patterns: dict[str, list[str]]) -> bool: - chunks = string_divide(test, 8) - search_str = "" - - for chunk in chunks: - if chunk in patterns["42"]: - search_str += "0" - elif chunk in patterns["31"]: - search_str += "1" - else: - return False - - # 42 = 0, 31 = 1 - res = search(r"^(0)+(1)+$", search_str) - return not (not res or search_str.count("0") <= search_str.count("1")) - - -def string_divide(string: str, div: int) -> list[str]: - l: list[str] = [] - for i in range(0, len(string), div): - l.append(string[i : i + div]) - return l - - -def main() -> None: - with open("input", "r") as f: - data = f.readlines() - - i = 0 - rules: list[tuple[str]] = [] - while data[i] != "\n": - rules.append(tuple(x for x in tuple(data[i].split(": ")))) - i += 1 - - tests: list[str] = [] - for line in data: - tests.append(line.strip()) - - patterns: dict[str, list[str]] = {} - - # Get "a" and "b" out of the way to do less comparisons in the next loop. Also make use of - # this loop to strip newlines off of all the rules. - for rule in rules: - rules[rules.index(rule)] = (rule[0], rule[1].strip()) - if len(rule[1]) == 4: - patterns[rule[0]] = [rule[1][1]] - - while len(patterns) != len(rules): - for rule in rules: - if rule[0] in patterns: - continue - - res = rule[1].split(" | ") - req_rules = list(OrderedDict.fromkeys(" ".join(res).split(" "))) - all_in = True - for req in req_rules: - if req not in patterns: - all_in = False - break - if all_in == True: - all_combos: list[str] = [] - for re in res: - re = re.split(" ") - if len(re) == 1: - all_combos.extend(patterns[re[0]]) - else: - for x in patterns[re[0]]: - for y in patterns[re[1]]: - all_combos.append(f"{x}{y}") - all_combos = list(OrderedDict.fromkeys(all_combos)) - - patterns[rule[0]] = all_combos - - print(len([test for test in tests if check(test, patterns)])) - - -if __name__ == "__main__": - main() |