85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
"""
|
||
|
|
Jour 05 du défi Advent Of Code pour l'année 2023
|
||
|
|
"""
|
||
|
|
|
||
|
|
def read_sample():
|
||
|
|
"""récupère les entrées depuis le fichier texte correspondant"""
|
||
|
|
with open('inputs/day05.txt', 'r') as f:
|
||
|
|
sample = f.read().split('\n\n')
|
||
|
|
sample = [ i for i in sample if i != '' ]
|
||
|
|
return sample
|
||
|
|
|
||
|
|
def parse_sample(sample):
|
||
|
|
data = []
|
||
|
|
data.append(("seeds", [int(i) for i in sample[0].split(" ")[1:] if i != ""]))
|
||
|
|
|
||
|
|
for subsample in sample[1:]:
|
||
|
|
d = subsample.split("\n")
|
||
|
|
name = d[0].split("-")[-1].split(" ")[0]
|
||
|
|
maps = [[int(j) for j in i.split(" ") if i != ""] for i in d[1:] if i != ""]
|
||
|
|
data.append((name, maps))
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
def get_locations(data, new_seed=False):
|
||
|
|
if not new_seed:
|
||
|
|
pos = [(i, 1) for i in data[0][1]]
|
||
|
|
else:
|
||
|
|
pos = []
|
||
|
|
for i in range(int(len(data[0][1])/2)):
|
||
|
|
pos.append((data[0][1][2*i], data[0][1][2*i+1]))
|
||
|
|
# print(pos)
|
||
|
|
|
||
|
|
for converter in data[1:]:
|
||
|
|
new_pos = []
|
||
|
|
for (source, rg) in pos:
|
||
|
|
found = False
|
||
|
|
min_dst = rg
|
||
|
|
for possible_dest in converter[1]:
|
||
|
|
cv_dest, cv_source, cv_range = possible_dest
|
||
|
|
|
||
|
|
if source >= cv_source and source <= cv_source+cv_range:
|
||
|
|
treated_range = min(rg, cv_range - (source - cv_source))
|
||
|
|
if treated_range > 0:
|
||
|
|
new_pos.append((source - (cv_source-cv_dest), treated_range))
|
||
|
|
|
||
|
|
#print(f"{converter[0]}: {source} -> {new_pos[-1]}")
|
||
|
|
if (rg > treated_range):
|
||
|
|
#print(f"using tr:", (source+treated_range, rg - treated_range))
|
||
|
|
pos.append((source+treated_range, rg - treated_range))
|
||
|
|
found = True
|
||
|
|
break
|
||
|
|
|
||
|
|
elif cv_source > source:
|
||
|
|
min_dst = min(min_dst, cv_source-source)
|
||
|
|
|
||
|
|
if not found:
|
||
|
|
new_pos.append((source, min_dst))
|
||
|
|
if min_dst < rg:
|
||
|
|
pos.append((source+min_dst, rg-min_dst))
|
||
|
|
#print(f"{converter[0]}: {source} -> {new_pos[-1]} (auto)")
|
||
|
|
pos = new_pos
|
||
|
|
return pos
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def part1(sample):
|
||
|
|
"""Partie 1 du défi"""
|
||
|
|
data = parse_sample(sample)
|
||
|
|
return min(get_locations(data))[0]
|
||
|
|
|
||
|
|
def part2(sample):
|
||
|
|
"""Partie 2 du défi"""
|
||
|
|
data = parse_sample(sample)
|
||
|
|
return min(get_locations(data, new_seed=True))[0]
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""Fonction principale"""
|
||
|
|
sample = read_sample()
|
||
|
|
print(f"part1: {part1(sample)}")
|
||
|
|
print(f"part2: {part2(sample)}")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|