結果

問題 No.2810 Have Another Go (Hard)
ユーザー hirayuu_ychirayuu_yc
提出日時 2024-07-06 15:01:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,646 ms / 3,000 ms
コード長 2,033 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 90,644 KB
最終ジャッジ日時 2024-07-06 15:03:08
合計ジャッジ時間 58,267 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 85 ms
76,800 KB
testcase_01 AC 1,036 ms
89,252 KB
testcase_02 AC 989 ms
89,636 KB
testcase_03 AC 1,020 ms
89,612 KB
testcase_04 AC 1,037 ms
89,512 KB
testcase_05 AC 1,133 ms
89,512 KB
testcase_06 AC 1,091 ms
89,764 KB
testcase_07 AC 1,011 ms
89,768 KB
testcase_08 AC 1,082 ms
89,348 KB
testcase_09 AC 1,106 ms
90,024 KB
testcase_10 AC 1,012 ms
89,904 KB
testcase_11 AC 127 ms
77,500 KB
testcase_12 AC 119 ms
77,912 KB
testcase_13 AC 103 ms
76,916 KB
testcase_14 AC 135 ms
78,524 KB
testcase_15 AC 161 ms
77,932 KB
testcase_16 AC 419 ms
80,284 KB
testcase_17 AC 1,277 ms
88,972 KB
testcase_18 AC 897 ms
84,468 KB
testcase_19 AC 1,256 ms
89,316 KB
testcase_20 AC 329 ms
79,612 KB
testcase_21 AC 515 ms
81,456 KB
testcase_22 AC 827 ms
84,196 KB
testcase_23 AC 678 ms
81,824 KB
testcase_24 AC 945 ms
85,648 KB
testcase_25 AC 1,425 ms
90,476 KB
testcase_26 AC 1,453 ms
89,964 KB
testcase_27 AC 1,520 ms
90,128 KB
testcase_28 AC 1,535 ms
90,500 KB
testcase_29 AC 1,583 ms
90,556 KB
testcase_30 AC 1,465 ms
90,092 KB
testcase_31 AC 1,541 ms
90,220 KB
testcase_32 AC 1,598 ms
90,640 KB
testcase_33 AC 1,646 ms
90,388 KB
testcase_34 AC 1,574 ms
90,508 KB
testcase_35 AC 1,532 ms
90,264 KB
testcase_36 AC 1,637 ms
90,380 KB
testcase_37 AC 1,524 ms
90,520 KB
testcase_38 AC 1,583 ms
90,644 KB
testcase_39 AC 1,525 ms
90,508 KB
testcase_40 AC 1,478 ms
90,344 KB
testcase_41 AC 1,423 ms
90,476 KB
testcase_42 AC 1,508 ms
90,388 KB
testcase_43 AC 1,567 ms
90,248 KB
testcase_44 AC 1,545 ms
90,564 KB
testcase_45 AC 1,641 ms
90,180 KB
testcase_46 AC 97 ms
76,912 KB
testcase_47 AC 123 ms
77,872 KB
testcase_48 AC 121 ms
77,832 KB
testcase_49 AC 110 ms
77,608 KB
testcase_50 AC 135 ms
77,688 KB
testcase_51 AC 313 ms
78,252 KB
testcase_52 AC 225 ms
77,852 KB
testcase_53 AC 380 ms
78,768 KB
testcase_54 AC 474 ms
78,368 KB
testcase_55 AC 240 ms
78,696 KB
testcase_56 AC 410 ms
78,240 KB
testcase_57 AC 326 ms
78,056 KB
testcase_58 AC 295 ms
77,916 KB
testcase_59 AC 522 ms
78,752 KB
testcase_60 AC 355 ms
78,300 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from functools import cache

MOD=998244353
def matrix_pow(a,x):
    ret=[0]*36
    for i in range(6):
        ret[i*7]=1
    while x>0:
        if x&1:
            nret=[0]*36
            for i in range(6):
                for k in range(6):
                    for j in range(6):
                        nret[i*6+j]+=ret[i*6+k]*a[k*6+j]
                        nret[i*6+j]%=MOD
            ret=nret
        na=[0]*36
        for i in range(6):
            for k in range(6):
                for j in range(6):
                    na[i*6+j]+=a[i*6+k]*a[k*6+j]
                    na[i*6+j]%=MOD
        a=na
        x>>=1
    return ret

@cache
def dice(x):
    if x<0:
        return 0
    return bostan_mori(x)

q=[]
Q=[1,-1,-1,-1,-1,-1,-1]
for i in range(64):
    q.append(Q)
    V=[0]*(len(Q)*2-1)
    for i in range(len(Q)):
        for j in range(len(Q)):
            if j%2==0:
                V[i+j]+=Q[i]*Q[j]
            else:
                V[i+j]-=Q[i]*Q[j]
            V[i+j]%=MOD
    Q=V[0::2]
    
def bostan_mori(N):
    P=[1]
    k=0
    while N>0:
        U=[0]*(len(P)+6)
        for i in range(len(P)):
            for j in range(7):
                if j%2==0:
                    U[i+j]+=P[i]*q[k][j]
                else:
                    U[i+j]-=P[i]*q[k][j]
                U[i+j]%=MOD
        P=U[N&1::2]
        N>>=1
        k+=1
    return P[0]%MOD

N,M,k=map(int,input().split())
C=list(map(int,input().split()))
lst=0
for i in range(6):
    lst+=dice(N*M-1-i)*(6-i)
lst%=MOD
for c in C:
    matrix=[0]*36
    for i in range(6):
        for j in range(6):
            if j==c:
                continue
            for k in range(6):
                if j+k+1>6:
                    continue
                matrix[i*6+j]+=(dice(N-k-1-i)-dice(c-i)*dice(N-k-1-c))
            matrix[i*6+j]%=MOD
    matrix=matrix_pow(matrix,M-1)
    ans=0
    for i in range(6):
        for j in range(6):
            ans+=matrix[i]*(dice(N-j-1-i)-dice(c-i)*dice(N-j-1-c))*(6-j)
        ans%=MOD
    print((lst-ans)%MOD)
0