結果

問題 No.840 ほむほむほむら
ユーザー pekempeypekempey
提出日時 2019-06-24 00:28:14
言語 OCaml
(5.1.0)
結果
AC  
実行時間 480 ms / 4,000 ms
コード長 1,220 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 20,480 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-09 00:44:43
合計ジャッジ時間 3,787 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 3 ms
6,816 KB
testcase_02 AC 11 ms
6,820 KB
testcase_03 AC 66 ms
6,820 KB
testcase_04 AC 1 ms
6,820 KB
testcase_05 AC 1 ms
6,820 KB
testcase_06 AC 3 ms
6,820 KB
testcase_07 AC 22 ms
6,820 KB
testcase_08 AC 121 ms
6,820 KB
testcase_09 AC 3 ms
6,820 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 5 ms
6,816 KB
testcase_12 AC 34 ms
6,816 KB
testcase_13 AC 316 ms
6,820 KB
testcase_14 AC 41 ms
6,820 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 7 ms
6,816 KB
testcase_17 AC 74 ms
6,816 KB
testcase_18 AC 396 ms
6,820 KB
testcase_19 AC 480 ms
6,816 KB
testcase_20 AC 1 ms
6,820 KB
testcase_21 AC 2 ms
6,820 KB
testcase_22 AC 7 ms
6,816 KB
testcase_23 AC 459 ms
6,816 KB
testcase_24 AC 7 ms
6,820 KB
testcase_25 AC 2 ms
6,816 KB
testcase_26 AC 11 ms
6,816 KB
testcase_27 AC 460 ms
6,820 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
File "Main.ml", line 1, characters 4-10:
1 | let [n; m] = read_line () |> String.split_on_char ' ' |> List.map int_of_string;;
        ^^^^^^
Warning 8 [partial-match]: this pattern-matching is not exhaustive.
Here is an example of a case that is not matched:
n::m::_::_

ソースコード

diff # 

let [n; m] = read_line () |> String.split_on_char ' ' |> List.map int_of_string;;

let s = m * m * m;;
let md = 998244353;;

let prod a b = begin
  let c = Array.make (s*s) 0 in
  for i = 0 to s - 1 do
    for k = 0 to s - 1 do
      for j = 0 to s - 1 do
        c.(i*s+j) <- (c.(i*s+j) + a.(i*s+k) * b.(k*s+j)) mod md
      done
    done
  done;
  c
end;;

let power a k = begin
  let res = ref (Array.make (s*s) 0) in
  for i = 0 to s - 1 do
    !res.(i*s+i) <- 1
  done;
  let a = ref a in
  let k = ref k in
  while !k > 0 do
    if !k mod 2 = 1 then
      res := prod !res !a;
    a := prod !a !a;
    k := !k / 2;
  done;
  !res
end;;

let a = Array.make (s*s) 0;;
let f i j k = let i = i mod m in
              let j = j mod m in
              let k = k mod m in i*m*m + j*m + k;;


let inc a i j = a.(i*s+j) <- a.(i*s+j) + 1;;

for i = 0 to m - 1 do
  for j = 0 to m - 1 do
    for k = 0 to m - 1 do
      inc a (f (i+1) j k) (f i j k);
      inc a (f i (j+i) k) (f i j k);
      inc a (f i j (k+j)) (f i j k);
    done
  done
done;;

let a = power a n;;
let ans = ref 0;;
for i = 0 to m - 1 do
  for j = 0 to m - 1 do
    ans := (!ans + a.(f i j 0)) mod md;
  done 
done;;

print_int !ans;;
print_newline ();;
0