Submission #2823730

---

Source Code Expand

let inf = 10101010101010
let rec iter a b f = if a >= b then () else (f a; iter (a+1) b f)
let q x = x * (x+1) / 2
let () =
  let width = 3000 in
  Scanf.scanf "%d %d %d" @@ fun r g b ->
  let a = Array.init 3 (fun _ -> Array.make width inf) in
  [(r, -100); (g, 0); (b, 100)] |> List.iteri (fun n (c, p) ->
    iter 0 width (fun i ->
      let s =
        let x = i - width/2 in
        if x+c <= p then q (p - x) - q (p - (x+c))
        else if x < p then q (p - x) + q ((x+c-1) - p)
        else q ((x+c-1) - p) - q ((x-1) - p)
      in
      if i+c < width then
        a.(n).(i+c) <-
          if i = 0 then inf
          else if n = 0 then min s a.(n).(i+c-1)
          else min (s + a.(n-1).(i)) a.(n).(i+c-1)));
  a.(2) |> Array.fold_left min 10101010101010
  |> Printf.printf "%d\n"

Submission Info

Judge Result