結果
問題 |
No.3213 depth max K
|
ユーザー |
|
提出日時 | 2025-07-25 23:11:50 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,903 bytes |
コンパイル時間 | 855 ms |
コンパイル使用メモリ | 90,864 KB |
実行使用メモリ | 15,944 KB |
最終ジャッジ日時 | 2025-07-25 23:12:32 |
合計ジャッジ時間 | 6,966 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 TLE * 7 |
ソースコード
// Original Python code: // n,k=map(int,input().split()) // M=998244353 // dp=[[0]*3 for _ in range(n+1)] // dp[0][0]=1 // for i in range(1,2*n+1): // ndp=[[0]*3 for _ in range(n+1)] // for j in range(n+1): // if j+1<=n: // if j+1==k: // ndp[j+1][1]+=sum(dp[j][:2])%M // elif j+1>k: // ndp[j+1][2]+=sum(dp[j])%M // else: // ndp[j+1][0]+=dp[j][0]%M // ndp[j+1][1]+=dp[j][1]%M // ndp[j+1][2]+=dp[j][2]%M // if j-1>=0: // ndp[j-1][0]+=dp[j][0]%M // ndp[j-1][1]+=dp[j][1]%M // ndp[j-1][2]+=dp[j][2]%M // dp=ndp // print(dp[0][1]%M) #include <iostream> #include <vector> using namespace std; int main() { int n, k; cin >> n >> k; const int M = 998244353; vector<vector<int>> dp(n + 1, vector<int>(3, 0)); dp[0][0] = 1; for (int i = 1; i <= 2 * n; ++i) { vector<vector<int>> ndp(n + 1, vector<int>(3, 0)); for (int j = 0; j <= n; ++j) { if (j + 1 <= n) { if (j + 1 == k) { int s = (dp[j][0] + dp[j][1]) % M; ndp[j + 1][1] = (ndp[j + 1][1] + s) % M; } else if (j + 1 > k) { int s = ((dp[j][0] + dp[j][1]) % M + dp[j][2]) % M; ndp[j + 1][2] = (ndp[j + 1][2] + s) % M; } else { ndp[j + 1][0] = (ndp[j + 1][0] + dp[j][0]) % M; ndp[j + 1][1] = (ndp[j + 1][1] + dp[j][1]) % M; ndp[j + 1][2] = (ndp[j + 1][2] + dp[j][2]) % M; } } if (j - 1 >= 0) { ndp[j - 1][0] = (ndp[j - 1][0] + dp[j][0]) % M; ndp[j - 1][1] = (ndp[j - 1][1] + dp[j][1]) % M; ndp[j - 1][2] = (ndp[j - 1][2] + dp[j][2]) % M; } } dp = move(ndp); } cout << dp[0][1] % M << endl; return 0; }