Submission #2850123

---

Source Code Expand

module M = Map.Make(struct type t = int let compare = (-) end)
let rec seq a b = if a > b then [] else a :: seq (a+1) b
let (>>=) x f = match x with Some x -> f x | _ -> None
let rec pt k b n =
  if k <= 0 then if n = 0 then [[]] else []
  else
    seq 1 (min n 3)
    |> List.map (fun l ->
      pt (k-1) (b+l) (n-l) |> List.map @@ fun e -> (b, l) :: e)
    |> List.concat

let fold_and f = List.fold_left (fun x y -> match x with None -> None | Some x -> f x y)
let find_some f = List.fold_left (fun a x -> match a with Some _ -> a | None -> f x) None

let () =
  Scanf.scanf "%d %d" @@ fun k n ->
  let vw = Array.init n (fun _ -> Scanf.scanf " %s %s" (fun v w -> v, w)) in
  let rec find idx m =
    if idx >= n then Some m else begin
    let v, w = vw.(idx) in
    pt (String.length v) 0 (String.length w) |> find_some (fun rs ->
      List.mapi (fun i (s, l) -> int_of_char v.[i] - 49, String.sub w s l) rs
      |> fold_and (fun m (i, s) ->
          try let s' = M.find i m in if s = s' then Some m else None
          with Not_found -> Some (M.add i s m)) (Some m)
      |> fun mm' -> mm' >>= find (idx+1))
    end
  in
  find 0 M.empty |> function Some m -> m |> M.iter (fun _ -> Printf.printf "%s\n")

Submission Info

Compile Error

File "./Main.ml", line 29, characters 20-82:
Warning 8: this pattern-matching is not exhaustive.
Here is an example of a value that is not matched:
None

Judge Result