summaryrefslogtreecommitdiff
path: root/06
diff options
context:
space:
mode:
Diffstat (limited to '06')
-rwxr-xr-x06/main.py23
1 files changed, 7 insertions, 16 deletions
diff --git a/06/main.py b/06/main.py
index 9951621..8418a4a 100755
--- a/06/main.py
+++ b/06/main.py
@@ -1,22 +1,13 @@
#!/bin/python3
import sys
+from math import sqrt
-times = [int(x) for x in input().split(":")[1].split()]
-distances = [int(x) for x in input().split(":")[1].split()]
-races = [(times[i], distances[i],) for i in range(len(times))]
+max_time = int(input().split(":")[1].replace(' ', ''))
+record_dist = int(input().split(":")[1].replace(' ', ''))
-def distance_for_race_len(race_len, hold_time):
- if hold_time <= 0: return 0
- if hold_time >= race_len: return 0
- return round((race_len - hold_time) * hold_time)
+lower_bound = ( max_time - sqrt( max_time ** 2 - 4 * record_dist ) ) / 2
+upper_bound = ( max_time + sqrt( max_time ** 2 - 4 * record_dist ) ) / 2
+possible_distances = round(abs(lower_bound - upper_bound))
-output = 1
-for max_time, record_dist in races:
- possible_distances = [distance_for_race_len(max_time, x) for x in range(max_time)]
- possible_distances = list(enumerate(possible_distances))
- possible_distances = [p for p in possible_distances if p[1] > record_dist]
-
- output *= len(possible_distances)
-
-print(f">> {output}")
+print(f">> {possible_distances}")