結果

問題 No.840 ほむほむほむら
ユーザー pekempeypekempey
提出日時 2019-06-15 03:56:48
言語 Crystal
(1.10.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 797 bytes
コンパイル時間 1,191 ms
コンパイル使用メモリ 171,404 KB
最終ジャッジ日時 2023-09-13 10:04:47
合計ジャッジ時間 2,651 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ(β)
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。(リジャッジ後のコンパイルエラーは公開されます)
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
Showing last frame. Use --error-trace for full trace.

In Main.cr:18:32

 18 | res = prod(res, a, s) if k.odd?
                                 ^---
Error: undefined method 'odd?' for Float64 (compile-time type is (Float64 | Int64))

ソースコード

diff #

MOD = 998244353_i64
n, m = gets.not_nil!.split.map(&.to_i64)
s = m*m*m

def prod(a, b, s)
  c = [0_i64] * (s*s)
  s.times do |i| s.times do |k| s.times do |j|
    c[i*s+j] += a[i*s+k] * b[k*s+j]
    c[i*s+j] %= MOD
  end end end
  c
end

def pow(a, k, s) 
  res = [0_i64] * (s*s)
  s.times{|i| res[i*s+i] = 1}
  while k > 0
    res = prod(res, a, s) if k.odd?
    a = prod(a, a, s)
    k /= 2
  end
  res
end

a = [0_i64] * (s*s)
f = ->(i : Int64, j : Int64, k : Int64) do
  i%m*m*m + j%m*m + k%m
end
m.times do |i| m.times do |j| m.times do |k|
  a[f.call(i,j,k)+f.call(i+1,j,k)*s] += 1
  a[f.call(i,j,k)+f.call(i,j+i,k)*s] += 1
  a[f.call(i,j,k)+f.call(i,j,k+j)*s] += 1
end end end

a = pow(a, n, s)
ans = 0_i64
m.times do |i| m.times do |j|
  ans += a[f.call(i,j,0_i64)]
end end
puts ans % MOD
0