|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +input=File.readlines(ARGV[0]) |
| 4 | + |
| 5 | +RULE=/^Step ([a-zA-Z]+) .* step ([a-zA-Z]+) can begin.$/.freeze |
| 6 | +input.map!{ |line|line.match(RULE).captures} |
| 7 | + |
| 8 | +dependents=input.group_by(&:first).transform_values{ |it|it.map(&:last)} |
| 9 | +dependencies=input.group_by(&:last).transform_values{ |it|it.map(&:first)} |
| 10 | + |
| 11 | +dependents.default_proc=proc{ |h,k|h[k]=[]} |
| 12 | +dependencies.default_proc=proc{ |h,k|h[k]=[]} |
| 13 | + |
| 14 | +steps_done=[] |
| 15 | +steps_current=dependents.keys.select{ |k|dependencies[k].empty?} |
| 16 | + |
| 17 | +workers=[] |
| 18 | +MAXWORKERS=5 |
| 19 | +elapsed=0 |
| 20 | + |
| 21 | +whilesteps_done.size <dependencies.size |
| 22 | +steps_current.sort! |
| 23 | +steps_current.uniq! |
| 24 | + |
| 25 | +is_ready=->(task){(dependencies[task] -steps_done).empty?} |
| 26 | +workers +=steps_current |
| 27 | +.select(&is_ready) |
| 28 | +.take(MAXWORKERS -workers.size) |
| 29 | +.map{ |sleight|[sleight,61 +('A'..'Z').find_index(sleight)]} |
| 30 | +.each{ |sleight|steps_current.delete(sleight[0])} |
| 31 | + |
| 32 | +min=workers.map(&:last).min |
| 33 | +workers.map!{ |worker,secs|[worker,secs -min]} |
| 34 | + |
| 35 | +workers.select{ |it|it[1].zero?}.eachdo |worker| |
| 36 | +workers.slice!(workers.index(worker)) |
| 37 | +steps_done <<worker[0] |
| 38 | +steps_current +=dependents[worker[0]] |
| 39 | +end |
| 40 | + |
| 41 | +elapsed +=min |
| 42 | +end |
| 43 | + |
| 44 | +putselapsed |
