2018/Day07/Day7B.py

83 lines
1.8 KiB
Python

import fileinput
from collections import defaultdict
from Day7A import build_graph
WORKERS = 5
BASE_TIME = 60
graph = build_graph()
order = []
to_consider = sorted(graph.keys())
worker_nodes = [None] * WORKERS
worker_times = [None] * WORKERS
active_workers = 0
time = 0
def work_seconds(node):
return ord(node) - ord('A') + BASE_TIME
def start_work(worker, node):
global active_workers
worker_nodes[worker] = node
worker_times[worker] = work_seconds(node)
active_workers += 1
def update_workers():
global time
global active_workers
for i in range(WORKERS):
if worker_nodes[i] is None:
continue
if worker_times[i]:
worker_times[i] -= 1
else:
active_workers -= 1
worker_nodes[i] = None
time += 1
def print_status():
worker_status = "".join(node if node else "." for node in worker_nodes)
todo = "".join(to_consider)
print(f"{time:>4d} {worker_status:<7} [{todo}]")
def main():
global active_workers
print("Time Workers To Do")
while to_consider or active_workers:
last_count = -1
while last_count != active_workers:
last_count = active_workers
for i, node in enumerate(to_consider):
parents = graph[node]
if node in worker_nodes:
continue
if any(parent in worker_nodes or parent in to_consider
for parent in parents):
continue
for w_id in range(WORKERS):
if worker_nodes[w_id] is None:
start_work(w_id, node)
break
else:
break
del to_consider[i]
continue
print_status()
update_workers()
print_status()
main()