結果

問題 No.840 ほむほむほむら
ユーザー pekempeypekempey
提出日時 2019-06-15 03:45:49
言語 Crystal
(1.11.2)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,043 bytes
コンパイル時間 1,876 ms
コンパイル使用メモリ 217,812 KB
最終ジャッジ日時 2024-04-27 02:52:45
合計ジャッジ時間 2,373 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

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

In Main.cr:26:32

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

ソースコード

diff # 

MOD = 998244353_i64
tmp = gets.not_nil!.split.map(&.to_i64)
n = tmp[0]
m = tmp[1]
s = m*m*m

def prod(a : Array(Int64), b : Array(Int64), s : Int64) : Array(Int64)
  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 : Array(Int64), k : Int64, s : Int64) : Array(Int64)
  res = [0_i64] * (s*s)
  s.times do |i|
    res[i*s+i] = 1
  end
  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
  j%=m
  k%=m
  i*m*m + j*m + k
end
m.times do |i|
  m.times do |j|
    m.times do |k|
      # ho
      a[f.call(i,j,k)+f.call(i+1,j,k)*s] += 1
      # mu
      a[f.call(i,j,k)+f.call(i,j+i,k)*s] += 1
      # ra
      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)]
    ans %= MOD
  end
end
puts ans
0