結果
問題 | No.1516 simple 門松列 problem Re:MASTER |
ユーザー | Kiri8128 |
提出日時 | 2021-05-21 22:57:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,359 ms / 6,000 ms |
コード長 | 1,247 bytes |
コンパイル時間 | 184 ms |
コンパイル使用メモリ | 82,168 KB |
実行使用メモリ | 82,388 KB |
最終ジャッジ日時 | 2024-04-18 16:30:25 |
合計ジャッジ時間 | 11,598 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
53,364 KB |
testcase_01 | AC | 138 ms
65,352 KB |
testcase_02 | AC | 994 ms
79,224 KB |
testcase_03 | AC | 44 ms
60,316 KB |
testcase_04 | AC | 43 ms
60,000 KB |
testcase_05 | AC | 48 ms
62,008 KB |
testcase_06 | AC | 78 ms
64,536 KB |
testcase_07 | AC | 172 ms
69,532 KB |
testcase_08 | AC | 265 ms
76,224 KB |
testcase_09 | AC | 44 ms
59,940 KB |
testcase_10 | AC | 73 ms
62,872 KB |
testcase_11 | AC | 46 ms
60,804 KB |
testcase_12 | AC | 45 ms
60,076 KB |
testcase_13 | AC | 44 ms
60,764 KB |
testcase_14 | AC | 2,044 ms
82,388 KB |
testcase_15 | AC | 980 ms
79,420 KB |
testcase_16 | AC | 503 ms
77,344 KB |
testcase_17 | AC | 210 ms
76,912 KB |
testcase_18 | AC | 86 ms
68,360 KB |
testcase_19 | AC | 62 ms
66,260 KB |
testcase_20 | AC | 2,359 ms
82,172 KB |
testcase_21 | AC | 2,306 ms
82,036 KB |
ソースコード
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]) % P, sum(a[KK:]) % P)