diff --git a/day_09/program.py b/day_09/program.py index aae4a10..17587c6 100755 --- a/day_09/program.py +++ b/day_09/program.py @@ -33,23 +33,29 @@ def get_disk(disk_map: str) -> tuple: def print_disk(disk: list) -> None: - for pos in disk: - if pos is None: + for block in disk: + if block is None: print(".", end="") else: - print(pos, end="") + print(block, end="") print() + +def defragment(disk: list, full_blocks: deque, free_blocks: deque) -> None: + while len(free_blocks) > 0: + pass + def main(): - disk_map = get_lines("sample-input.txt")[0] - # disk_map = get_lines("sample-input2.txt")[0] + # disk_map = get_lines("sample-input.txt")[0] + disk_map = get_lines("sample-input2.txt")[0] # disk_map = get_lines("input.txt")[0] disk, free_blocks, full_blocks = get_disk(disk_map) print_disk(disk) - print(free_blocks) - print(full_blocks) - + # print(free_blocks) + # print(full_blocks) + defragment(disk, free_blocks, full_blocks) + if __name__ == '__main__': main()