cosmetics

Author: anight

2023/day05/run.py

@@ -9,7 +9,7 @@ def load_data(filename):
 			if line.startswith('seeds:'):
 				seeds = list(map(int, line.split(': ')[1].split()))
 			elif line.endswith(':'):
-				map_from, _, map_to = line[:-len(" map:")].split('-')
+				map_from, map_to = line[:-len(" map:")].split('-to-')
 			elif line == '':
 				if len(current_map) > 0:
 					maps[ (map_from, map_to) ] = current_map
@@ -35,8 +35,6 @@ def load_data(filename):
 			if src <= value < src + r:
 				value = dst + value - src
 				break
-		else:
-			pass
 		key = next_key
 	min_location = min(min_location, value)
 
@@ -50,7 +48,8 @@ def load_data(filename):
 min_location = float('inf')
 
 for i in range(0, len(seeds), 2):
-	ranges_from = P.closed(seeds[i], seeds[i] + seeds[i+1] - 1)
+	src, r = seeds[i:i+2]
+	ranges_from = P.closed(src, src + r - 1)
 	key = 'seed'
 	while key != 'location':
 		translated = P.empty()

2023/day06/run.py

@@ -4,7 +4,7 @@ def load_data(filename):
 	with open(filename, 'r') as f:
 		times = f.readline().rstrip('\n').split()[1:]
 		distances = f.readline().rstrip('\n').split()[1:]
-		return map(int, times), map(int, distances)
+		return list(map(int, times)), list(map(int, distances))
 
 # Part One
 
@@ -17,15 +17,13 @@ def distance(total_time, button_time):
 	return (total_time - button_time) * speed
 
 for t, record_distance in zip(times, distances):
-	variants = [ 1 for b in range(t+1) if distance(t, b) > record_distance ]
-	total *= len(variants)
+	variants = sum( 1 for b in range(t+1) if distance(t, b) > record_distance )
+	total *= variants
 
 print(total)
 
 # Part Two
 
-times, distances = load_data('input.txt')
-
 time = int(''.join(map(str, times)))
 record_distance = int(''.join(map(str, distances)))
 
@@ -37,7 +35,7 @@ def search(low, high, f):
 			low = mid
 		else:
 			high = mid
-	return mid
+	return high
 
 def lower_bound(x):
 	if distance(time, x) <= record_distance: