結果

問題 No.1366 交換門松列・梅
ユーザー esehara
提出日時 2021-06-01 21:02:05
言語 OCaml
(5.2.1)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,218 bytes
コンパイル時間 651 ms
コンパイル使用メモリ 20,636 KB
実行使用メモリ 6,272 KB
最終ジャッジ日時 2025-06-20 11:02:53
合計ジャッジ時間 1,268 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff #

let check_umezawa ar =
  ((ar.(0) <> ar.(1)) && (ar.(1) <> ar.(2)) && (ar.(0) <> ar.(2))) &&
  (((ar.(0) < ar.(1)) && (ar.(2) < ar.(1))) || ((ar.(0) > ar.(1)) && (ar.(2) > ar.(1))));;

let split_on_char sep s =
  let r = ref [] in
  let j = ref (String.length s) in
  for i = String.length s - 1 downto 0 do
    if String.unsafe_get s i = sep then begin
      r := String.sub s (i + 1) (!j - i - 1) :: !r;
      j := i
    end
  done;
  String.sub s 0 !j :: !r;;

let rec swap ar1 ar2 a b =
  if (b > 2) then begin print_string "No"; print_newline (); end
  else if (a > 2) then swap ar1 ar2 0 (b + 1)
  else
    let copy_ar1 = Array.copy ar1 in
    let copy_ar2 = Array.copy ar2 in
    let target_a = ar1.(a) in
    let target_b = ar2.(b) in
    begin 
      copy_ar1.(a) <- target_b; copy_ar2.(b) <- target_a;
      if (check_umezawa copy_ar1) && (check_umezawa copy_ar2)
      then
        begin
          print_string "Yes"; print_newline ();
        end
      else swap ar1 ar2 (a + 1) b
    end;;

let () =
  let ar1 = read_line () |> split_on_char ' ' |> List.map int_of_string |> Array.of_list in
  let ar2 = read_line () |> split_on_char ' ' |> List.map int_of_string |> Array.of_list in
  swap ar1 ar2 0 0;;
0