diff options
Diffstat (limited to 'day7/__init__.py')
| -rw-r--r-- | day7/__init__.py | 5 |
1 files changed, 3 insertions, 2 deletions
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): | |||
| 37 | @staticmethod | 37 | @staticmethod |
| 38 | def calculate_cost(start, end) -> int: | 38 | def calculate_cost(start, end) -> int: |
| 39 | base_cost = AssignmentOne.calculate_cost(start, end) | 39 | base_cost = AssignmentOne.calculate_cost(start, end) |
| 40 | extra = sum(i for i in range(base_cost)) | 40 | # extra = sum(i for i in range(base_cost)) |
| 41 | return base_cost + extra \ No newline at end of file | 41 | # return base_cost + extra |
| 42 | return int(base_cost * (base_cost + 1) / 2) | ||