結果

問題 No.840 ほむほむほむら
ユーザー pekempey
提出日時 2019-06-24 00:28:14
言語 OCaml
(5.2.1)
結果
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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
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