結果
問題 | No.1513 simple 門松列 problem |
ユーザー | SSRS |
提出日時 | 2021-05-21 21:30:56 |
言語 | C++14 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 182 ms / 3,000 ms |
コード長 | 1,784 bytes |
コンパイル時間 | 2,209 ms |
コンパイル使用メモリ | 172,520 KB |
実行使用メモリ | 131,328 KB |
最終ジャッジ日時 | 2024-04-18 14:36:31 |
合計ジャッジ時間 | 3,505 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 182 ms
131,328 KB |
testcase_03 | AC | 2 ms
6,940 KB |
testcase_04 | AC | 2 ms
6,940 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 2 ms
6,940 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,940 KB |
testcase_09 | AC | 2 ms
6,940 KB |
testcase_10 | AC | 2 ms
6,944 KB |
testcase_11 | AC | 2 ms
6,944 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 2 ms
6,940 KB |
testcase_14 | AC | 177 ms
128,512 KB |
testcase_15 | AC | 176 ms
129,152 KB |
testcase_16 | AC | 180 ms
130,560 KB |
testcase_17 | AC | 110 ms
82,816 KB |
testcase_18 | AC | 39 ms
29,696 KB |
testcase_19 | AC | 4 ms
6,940 KB |
testcase_20 | AC | 22 ms
19,584 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; int main(){ int N, K; cin >> N >> K; vector<vector<vector<long long>>> dp1(N - 1, vector<vector<long long>>(K, vector<long long>(K, 0))); vector<vector<vector<long long>>> dp2(N - 1, vector<vector<long long>>(K, vector<long long>(K, 0))); for (int i = 0; i < K; i++){ for (int j = 0; j < K; j++){ if (i != j){ dp1[0][i][j]++; dp2[0][i][j] += i + j; } } } for (int i = 0; i < N - 2; i++){ for (int k = 0; k < K; k++){ long long sum11 = 0, sum12 = 0; for (int j = 0; j < k; j++){ sum11 += dp1[i][j][k]; sum12 += dp2[i][j][k]; } for (int j = 0; j < k; j++){ dp1[i + 1][k][j] += sum11; dp1[i + 1][k][j] += MOD - dp1[i][j][k]; dp1[i + 1][k][j] %= MOD; dp2[i + 1][k][j] += sum12 + k * sum11; dp2[i + 1][k][j] += MOD - dp2[i][j][k]; dp2[i + 1][k][j] += MOD - dp1[i][j][k] * k % MOD; dp2[i + 1][k][j] %= MOD; } long long sum21 = 0, sum22 = 0; for (int j = k + 1; j < K; j++){ sum21 += dp1[i][j][k]; sum22 += dp2[i][j][k]; } for (int j = k + 1; j < K; j++){ dp1[i + 1][k][j] += sum21; dp1[i + 1][k][j] += MOD - dp1[i][j][k]; dp1[i + 1][k][j] %= MOD; dp2[i + 1][k][j] += sum22 + k * sum21; dp2[i + 1][k][j] += MOD - dp2[i][j][k]; dp2[i + 1][k][j] += MOD - dp1[i][j][k] * k % MOD; dp2[i + 1][k][j] %= MOD; } } } long long ans1 = 0, ans2 = 0; for (int i = 0; i < K; i++){ for (int j = 0; j < K; j++){ ans1 += dp1[N - 2][i][j]; ans2 += dp2[N - 2][i][j]; } } ans1 %= MOD; ans2 %= MOD; cout << ans1 << ' ' << ans2 << endl; }