From d22c1b117eda357267f2c93193d23d32779cbd69 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Tue, 7 Dec 2021 16:20:04 +0100 Subject: Made it faster --- day7/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'day7/__init__.py') diff --git a/day7/__init__.py b/day7/__init__.py index b3579b7..933b9e0 100644 --- a/day7/__init__.py +++ b/day7/__init__.py @@ -37,5 +37,6 @@ class AssignmentTwo(Assignment): @staticmethod def calculate_cost(start, end) -> int: base_cost = AssignmentOne.calculate_cost(start, end) - extra = sum(i for i in range(base_cost)) - return base_cost + extra \ No newline at end of file + # extra = sum(i for i in range(base_cost)) + # return base_cost + extra + return int(base_cost * (base_cost + 1) / 2) -- cgit v1.2.3