結果

問題 No.2944 Sigma Partition Problem
コンテスト
ユーザー a1048576
提出日時 2024-10-18 22:01:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 90 ms / 4,000 ms
コード長 630 bytes
コンパイル時間 4,093 ms
コンパイル使用メモリ 234,108 KB
実行使用メモリ 66,048 KB
最終ジャッジ日時 2024-10-18 22:44:24
合計ジャッジ時間 7,316 ms
ジャッジサーバーID
（参考情報） 		judge7 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all> 
using mint = atcoder::modint998244353;
#define int long long
signed main() {
  int Q;
  cin >> Q;
  int m = 4000;
  vector<vector<mint>> dp(m+1,vector<mint>(m+1,0));
  dp[0][0] = 1;
  for(int i = 0; i < m; i++) {
    dp[i+1] = dp[i];
    for(int j = 0; j <= m; j++) {
      if(i+1+j <= m) dp[i+1][j+i+1]+=dp[i+1][j];
    }
  }
  while(Q--) {
    int t,n,k;
    cin >> t >> n >> k;
    if(t == 1) {
      mint ans = dp[k][n];
      cout << ans.val() << endl;
    }
    else {
      mint ans = 0;
      ans = dp[k][n];
      cout << ans.val() << endl;
    }
  }
}
0