結果
問題 | No.2944 Sigma Partition Problem |
ユーザー |
![]() |
提出日時 | 2024-10-18 22:32:30 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 272 ms / 4,000 ms |
コード長 | 1,310 bytes |
コンパイル時間 | 3,318 ms |
コンパイル使用メモリ | 271,708 KB |
実行使用メモリ | 128,840 KB |
最終ジャッジ日時 | 2024-10-18 22:52:15 |
合計ジャッジ時間 | 10,388 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 24 |
ソースコード
#include<bits/stdc++.h> namespace { #pragma GCC diagnostic ignored "-Wunused-function" #include<atcoder/all> #pragma GCC diagnostic warning "-Wunused-function" using namespace std; using namespace atcoder; #define rep(i,n) for(int i = 0; i < (int)(n); i++) #define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--) #define all(x) begin(x), end(x) #define rall(x) rbegin(x), rend(x) template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; } template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; } using ll = long long; using P = pair<int,int>; using VI = vector<int>; using VVI = vector<VI>; using VL = vector<ll>; using VVL = vector<VL>; using mint = modint998244353; constexpr int MX = 4005; mint M[MX][MX], L[MX][MX]; } int main() { ios::sync_with_stdio(false); cin.tie(0); { mint dp[MX]; dp[0] = 1; for (int k = 1; k < MX; k++) { rep(i, MX - k) dp[i+k] += dp[i]; rep(i, MX) M[i][k] = dp[i]; } } { L[0][0] = 1; rep(i, MX) rep(j, MX) { if (i) L[i][j] += L[i-1][j]; if (j - i >= 0) L[i][j] += L[i][j-i]; } } int q; cin >> q; while (q--) { int type, n, k; cin >> type >> n >> k; cout << (type ? M : L)[n][k].val() << '\n'; } }