more ocaml, more to read

(* vim: set ts=2 sw=2 et : *)

type square = Unknown | NotYet | Black | White | Both

let name_from_square = function
  Unknown -> "Unknown" | NotYet -> "NotYet" | Black -> "Black"
| White -> "White" | Both -> "Both"

let read_board file =
  let board = Array.make_matrix 19 19 Unknown in
  for y = 0 to 18 do
    let line = input_line file in
    for x = 0 to 18 do
      let c = line.[x*2] in
      let newsquare = (match c with
                        '+' | '*' -> Unknown
                      | 'O' -> White
                      | '@' -> Black
                      | _ -> raise (Invalid_argument (Char.escaped c))) in
      board.(x).(y) <- newsquare;
    done;
  done;
  board
;;

let output_board board =
  for y = 0 to 18 do
    for x = 0 to 18 do
      let c = (match board.(x).(y) with
        Unknown -> "+"
      | Both -> "?"
      | White -> "O" | Black -> "@" | NotYet -> "#") in
      print_string (c ^ " ");
    done;
    print_newline ();
  done;
  print_newline ();
  ()
;;

let count (a,b,c,d) color =
  if a = color then 1 else 0 +
  if b = color then 1 else 0 +
  if c = color then 1 else 0 +
  if d = color then 1 else 0

let rec scan board x y =
  (*Printf.printf "scanning from %d %d\n" x y;
  output_board board;*)
  let s = board.(x).(y) in
  match s with
    Black | White | Both | NotYet -> s
  | Unknown ->
    board.(x).(y) <- NotYet;
    let l = if x > 0 then scan board (x-1) y else Unknown in
    let r = if y > 0 then scan board x (y-1) else Unknown in
    let u = if x < 18 then scan board (x+1) y else Unknown in
    let d = if y < 18 then scan board x (y+1) else Unknown in
    let both = count (l,r,u,d) Both in
    if both > 0 then Both
    else
      let white = count (l,r,u,d) White in
      let black = count (l,r,u,d) Black in
      if black > 0 && white = 0 then Black
      else if white > 0 && black = 0 then White
      else if white > 0 && black > 0 then Both
      else Unknown

let rec fill_area board x y s =
  match board.(x).(y) with
    Black | White | Both -> ()
  | Unknown | NotYet ->
    board.(x).(y) <- s;
    if x > 0 then fill_area board (x-1) y s;
    if y > 0 then fill_area board x (y-1) s;
    if x < 18 then fill_area board (x+1) y s;
    if y < 18 then fill_area board x (y+1) s;
    ()

let fixup_board board =
  for y = 0 to 18 do
    for x = 0 to 18 do
      if board.(x).(y) = Unknown then begin
        let s = scan board x y in
        fill_area board x y s;
      end;
    done;
  done;
  ()

let count_board board =
  let black = ref 0 and white = ref 0 in
  for y = 0 to 18 do
    for x = 0 to 18 do
      match board.(x).(y) with
        Black -> black := !black + 1
      | White -> white := !white + 1
      | _ -> ()
    done
  done;
  (!black, !white)
;;

let run_board file =
  try while true do
    let _ = input_line file in
    let board = read_board file in
    fixup_board board;
    let (black, white) = count_board board in
      Printf.printf "%d %d\n" black white
  done with End_of_file -> (); ()
;;

let file = open_in "input" in
  run_board file;
  run_board file