ESM オフラインリアルタイムどう書くの問題を解いてみた

def try_sit( chairs, cus )
  n=cus[0]
  pos = 8.times.find{ |x| (x...(x+n)).all?{ |y| chairs[y%8]==0 } }
  if pos
    (pos...(pos+n)).each{ |x| chairs[x%8]=1 }
    cus.shift
  end
end

def progress( chairs )
  chairs.size.times do |ix|
    chairs[ix] = (chairs[ix]+1) % 4 if chairs[ix]!=0
  end
end

def solve_impl( chairs, cus )
  loop do
    try_sit( chairs, cus )
    return chairs if cus.empty?
    progress(chairs)
  end
end

def solve(s)
  ch=solve_impl( [0]*8, s.chars.map( &:to_i ) )
  ch.map{ |x| x==0 ? 0 : 1 }.join
end

DATA.map do |line|
  num, src, expected = line.split( /\s+/ )
  actual = solve( src )
  ok = actual == expected
  puts [num, (ok ? "ok" : "***NG***"), src, expected, actual].join(" ")
  ok
end.all?{ |x| puts( x ? "all okay!" : "some failed" ) }

__END__
1 3316  11111110
2 3342153 11111111