結果

問題 No.2944 Sigma Partition Problem
コンテスト
ユーザー 横山緑
提出日時 2024-10-18 23:00:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 582 ms / 4,000 ms
コード長 4,518 bytes
コンパイル時間 6,948 ms
コンパイル使用メモリ 336,772 KB
実行使用メモリ 199,168 KB
最終ジャッジ日時 2024-10-18 23:02:04
合計ジャッジ時間 22,712 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 2 "library/template.hpp"
// #pragma GCC target("avx2")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")

#include <bits/stdc++.h>
using namespace std;
// https://xn--kst.jp/blog/2019/08/29/cpp-comp/
// debug methods
// usage: debug(x,y);
#define CHOOSE(a) CHOOSE2 a
#define CHOOSE2(a0, a1, a2, a3, a4, x, ...) x
#define debug_1(x1) cout << #x1 << ": " << x1 << endl
#define debug_2(x1, x2)                                                        \
    cout << #x1 << ": " << x1 << ", " #x2 << ": " << x2 << endl
#define debug_3(x1, x2, x3)                                                    \
    cout << #x1 << ": " << x1 << ", " #x2 << ": " << x2 << ", " #x3 << ": "    \
         << x3 << endl
#define debug_4(x1, x2, x3, x4)                                                \
    cout << #x1 << ": " << x1 << ", " #x2 << ": " << x2 << ", " #x3 << ": "    \
         << x3 << ", " #x4 << ": " << x4 << endl
#define debug_5(x1, x2, x3, x4, x5)                                            \
    cout << #x1 << ": " << x1 << ", " #x2 << ": " << x2 << ", " #x3 << ": "    \
         << x3 << ", " #x4 << ": " << x4 << ", " #x5 << ": " << x5 << endl
#ifdef LOCAL
#define debug(...)                                                             \
    CHOOSE((__VA_ARGS__, debug_5, debug_4, debug_3, debug_2, debug_1, ~))      \
    (__VA_ARGS__)
#else
#define debug(...)
#endif

using ll = long long;
using vl = vector<ll>;
using Graph = vector<vector<ll>>;
using P = pair<ll, ll>;
#define all(v) v.begin(), v.end()
template <typename T> inline bool chmax(T &a, T b) {
    return ((a < b) ? (a = b, true) : (false));
}
template <typename T> inline bool chmin(T &a, T b) {
    return ((a > b) ? (a = b, true) : (false));
}
#define rep1(i, n) for(ll i = 1; i <= ((ll)n); ++i)
// https://trap.jp/post/1224/
template <class... T> constexpr auto min(T... a) {
    return min(initializer_list<common_type_t<T...>>{a...});
}
template <class... T> constexpr auto max(T... a) {
    return max(initializer_list<common_type_t<T...>>{a...});
}
template <class... T> void input(T &...a) { (cin >> ... >> a); }
template <class T> void input(vector<T> &a) {
    for(T &x : a)
        cin >> x;
}
void print() { cout << '\n'; }
template <class T, class... Ts> void print(const T &a, const Ts &...b) {
    cout << a;
    (cout << ... << (cout << ' ', b));
    cout << '\n';
}
template <class T> void print(vector<T> &x) {
    if(x.size()) {
        for(ll i = 0; i < x.size(); i++) {
            cout << x[i] << " \n"[i + 1 == x.size()];
        }
    }
}
#define INT(...)                                                               \
    int __VA_ARGS__;                                                           \
    input(__VA_ARGS__)
#define LL(...)                                                                \
    long long __VA_ARGS__;                                                     \
    input(__VA_ARGS__)
#define STR(...)                                                               \
    string __VA_ARGS__;                                                        \
    input(__VA_ARGS__)
#define REP1(a) for(ll i = 0; i < a; i++)
#define REP2(i, a) for(ll i = 0; i < a; i++)
#define REP3(i, a, b) for(ll i = a; i < b; i++)
#define REP4(i, a, b, c) for(ll i = a; i < b; i += c)
#define overload4(a, b, c, d, e, ...) e
#define rep(...) overload4(__VA_ARGS__, REP4, REP3, REP2, REP1)(__VA_ARGS__)

ll inf = 3e18;
vl dx = {1, -1, 0, 0};
vl dy = {0, 0, 1, -1};
#line 2 "yukicoder/450/test.cpp"
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
void solve() {
    LL(q);
    ll max = 5000;
    vector dp(max + 1, vector<mint>(max + 1, 0));
    dp[0][0] = 1;
    rep(i, max + 1) {
        rep1(j, max) {
            if(i >= j)
                dp[i][j] += dp[i - j][j];
        }
        rep(j, max) {
            dp[i][j + 1] += dp[i][j];
        }
    }
    vector pd(max + 1, vector<mint>(max + 1, 0));
    pd[0][0] = 1;
    rep(i,max){
        dp[0][i] = 1;
        dp[i][1] = 1;
    }
    rep(i, max + 1) {
        rep(j, max + 1) {
            if(i >= j && j>=1)
                pd[i][j] = pd[i - j][j] + pd[i][j - 1];
        }
    }
    rep(_, q) {
        LL(flag, n, k);
        if(flag == 1) {
            print(dp[n][k].val());
        } else {
            print(dp[n][k].val());
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    cout << std::setprecision(16);
    int t = 1;
    rep(_, t) {
        solve();
    }
}
0