結果

問題 No.1516 simple 門松列 problem Re:MASTER
コンテスト
ユーザー Kiri8128
提出日時 2021-05-21 22:56:49
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,239 bytes
コンパイル時間 491 ms
コンパイル使用メモリ 82,288 KB
実行使用メモリ 82,704 KB
最終ジャッジ日時 2024-10-10 09:41:35
合計ジャッジ時間 11,570 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 8 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

def mmult(A, B):
    global mod
    n, m, l = len(A), len(B), len(B[0])
    ret = [[0] * l for _ in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(l):
                ret[i][k] = (ret[i][k] + A[i][j] * B[j][k]) % mod
    return ret
def mpow(A, n):
    if n == 1: return A
    if n == 0: return [[1 if i == j else 0 for j in range(len(A))] for i in range(len(A))]
    return mmult(mpow(A, n - 1), A) if n % 2 else mpow(mmult(A, A), n // 2)

mod = 998244353

P = 998244353
N, K = map(int, input().split())
KK = K ** 2
L = 2 * KK
M = [[0] * L for _ in range(L)]
A = [0] * L

X = [[1] * (K + 1) for _ in range(K)]
for i in range(K):
    for j in range(K):
        if i == j: continue
        A[i*K+j] = 1
        A[i*K+j+KK] = i + j

for i in range(K):
    for j in range(i):
        for k in range(j+1, K):
            if k == i: continue
            M[i*K+j][j*K+k] = 1
            M[i*K+j+KK][j*K+k+KK] = 1
            M[i*K+j][j*K+k+KK] = k
    for j in range(i+1, K):
        for k in range(j):
            if k == i: continue
            M[i*K+j][j*K+k] = 1
            M[i*K+j+KK][j*K+k+KK] = 1
            M[i*K+j][j*K+k+KK] = k

a = mmult([A], mpow(M, N - 2))[0]
print(sum(a[:KK]), sum(a[KK:]))
0