diff --git a/day11/part1.py b/day11/part1.py index aeafae3..195717b 100644 --- a/day11/part1.py +++ b/day11/part1.py @@ -7,9 +7,7 @@ def create_op(op_line): if "old * old" in op_line: return lambda x: x**2 val = int(op_line[6:]) - if "*" in op_line: - return lambda x: x*val - return lambda x: x+val + return (lambda x: x*val) if "*" in op_line else (lambda x: x+val) for line in lines: diff --git a/day11/part2.py b/day11/part2.py index 3147eb2..f8de387 100644 --- a/day11/part2.py +++ b/day11/part2.py @@ -9,9 +9,7 @@ def create_op(op_line): if "old * old" in op_line: return lambda x: x**2 val = int(op_line[6:]) - if "*" in op_line: - return lambda x: x*val - return lambda x: x+val + return (lambda x: x*val) if "*" in op_line else (lambda x: x+val) for line in lines: