結果

問題 No.3044 よくあるカエルさん
ユーザー sasa8uyauya
提出日時 2025-03-01 01:41:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 418 ms / 2,000 ms
コード長 556 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 76,584 KB
最終ジャッジ日時 2025-03-01 01:41:07
合計ジャッジ時間 5,091 ms
ジャッジサーバーID
(参考情報)
judge4 / judge6
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

n,t=map(int,input().split())
k,l=map(int,input().split())
M=998244353
L=t

def seki(a,b):
  c=[[0]*L for i in range(L)]
  for i in range(L):
    for j in range(L):
      for k in range(L):
        c[i][j]+=a[i][k]*b[k][j]
        c[i][j]%=M
  return c

A=[[0]*L for i in range(L)]
A[0][1-1]=(k-1)*pow(6,M-2,M)%M
A[0][2-1]=(l-k)*pow(6,M-2,M)%M
A[0][t-1]=(6-l+1)*pow(6,M-2,M)%M
for i in range(L-1):
  A[i+1][i]=1

B=[[0]*L for i in range(L)]
for i in range(L):
  B[i][i]=1

for i in range(30):
  if ((n-1)>>i)&1:
    B=seki(A,B)
  A=seki(A,A)

print(B[0][0])
0