結果

問題 No.429 CupShuffle
コンテスト
ユーザー ichibanshibori
提出日時 2016-10-05 15:19:04
言語 OCaml
(5.2.1)
結果
AC  
実行時間 131 ms / 2,000 ms
コード長 1,761 bytes
コンパイル時間 330 ms
コンパイル使用メモリ 20,096 KB
実行使用メモリ 17,016 KB
最終ジャッジ日時 2024-10-08 23:50:53
合計ジャッジ時間 1,900 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

let ( @! ) lst n = List.nth lst n

let read_AB_arr k x =
  let abArr = Array.make k (0, 0) in
  let rec read_AB_arr' i =
    if i >= k then abArr
    else (
      let line = read_line () in
      let a, b =
      	if i = (x - 1) then (-1, -1)
        else
          Str.split (Str.regexp_string " ") line |>
          List.map int_of_string |>
          (fun lst -> (lst @! 0, lst @! 1))
      in
      abArr.(i) <- (a, b);
      read_AB_arr' (i + 1))
  in
  read_AB_arr' 0

let do_swap_cup (cupArr: int array) abArr startIdx step =
  let swap_cup a b =
    let tmp = cupArr.(a - 1) in
    cupArr.(a - 1) <- cupArr.(b - 1);
    cupArr.(b - 1) <- tmp
  in
  let rec do_swap_cup' idx =
  	let a, b = abArr.(idx) in
    if a <> -1 then (
      swap_cup a b;
      do_swap_cup' (idx + step))
  in
  do_swap_cup' startIdx

let diff_cups n arr1 arr2 =
  let rec diff_cups' idx cnt result =
    if idx >= n || cnt = 2 then result
    else (
      let nextCnt, nextResult =
        if arr1.(idx) <> arr2.(idx) then
          (cnt + 1), (idx + 1) :: result
        else cnt, result
      in
      diff_cups' (idx + 1) nextCnt nextResult)
  in
  diff_cups' 0 0 [] |>
  List.rev

let () =
  let n, k, x = read_line () |>
                Str.split (Str.regexp_string " ") |>
                List.map int_of_string |>
                (fun lst -> (lst @! 0, lst @! 1, lst @! 2))
  in
  let abArr = read_AB_arr k x
  and cArr = read_line () |>
             Str.split (Str.regexp_string " ") |>
             List.map int_of_string |>
             Array.of_list
  and cupArr = Array.init n (fun i -> i + 1)
  in
  do_swap_cup cupArr abArr 0 1;
  do_swap_cup cArr abArr (k - 1) (-1);
  diff_cups n cupArr cArr |>
  List.map string_of_int |>
  String.concat " " |>
  print_endline
0