heiko/advent_of_code_2024
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

incomplete state of day 9 - step 2

Browse source
This commit is contained in:
Heiko Ludwig 2024-12-10 17:41:34 +01:00
parent 11848d268b
commit b7ea34fd2b
1 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

32
day_09/program.py
View file

@ -12,30 +12,44 @@ def get_lines(filename: str) -> list:
def get_disk(disk_map: str) -> tuple: def get_disk(disk_map: str) -> tuple:
disk = [] disk = []
free_blocks = deque()
full_blocks = deque() full_blocks = deque()
free_blocks = deque()
file_id = 0 file_id = 0
head_pos = 0
for pos, count_str in enumerate(disk_map): for pos, count_str in enumerate(disk_map):
count = int(count_str) count = int(count_str)
for i in range(count): if pos % 2 == 0:
if pos % 2 == 0: for i in range(count):
disk.append(file_id) disk.append(file_id)
else: full_blocks.append(head_pos)
head_pos += 1
file_id += 1
else:
for i in range(count):
disk.append(None) disk.append(None)
free_blocks.append(head_pos)
head_pos += 1
return disk, free_blocks, full_blocks return disk, free_blocks, full_blocks
def print_disk(disk: list) -> None: def print_disk(disk: list) -> None:
pass for pos in disk:
if pos is None:
print(".", end="")
else:
print(pos, end="")
print()
def main(): def main():
# disk_map = get_lines("sample-input.txt")[0] disk_map = get_lines("sample-input.txt")[0]
disk_map = get_lines("sample-input2.txt")[0] # disk_map = get_lines("sample-input2.txt")[0]
# disk_map = get_lines("input.txt")[0] # disk_map = get_lines("input.txt")[0]
disk, free_blocks, full_blocks = get_disk(disk_map) disk, free_blocks, full_blocks = get_disk(disk_map)
print(disk) print_disk(disk)
print(free_blocks)
print(full_blocks)
if __name__ == '__main__': if __name__ == '__main__':
main() main()