結果

問題 No.1956 猫の額
ユーザー SSRS
提出日時 2022-05-23 21:06:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 988 bytes
コンパイル時間 2,726 ms
コンパイル使用メモリ 190,056 KB
実行使用メモリ 74,920 KB
最終ジャッジ日時 2024-09-20 13:57:26
合計ジャッジ時間 10,650 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 5 MLE * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
int main(){
  int N, M, C;
  cin >> N >> M >> C;
  vector<int> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  int S = 0;
  for (int i = 0; i < N; i++){
    S += A[i];
  }
  vector<vector<int>> dp(N + 1, vector<int>(S + 1, 0));
  vector<vector<int>> dp2(N + 1, vector<int>(S + 1, 0));
  dp[0][0] = 1;
  int T = 0;
  for (int i = 0; i < N; i++){
    for (int j = 0; j <= i + 1; j++){
      for (int k = 0; k <= T + A[i]; k++){
        dp2[j][k] = 0;
      }
    }
    for (int j = 0; j <= i; j++){
      for (int k = 0; k <= T; k++){
        dp2[j][k] += dp[j][k];
        if (dp2[j][k] >= M){
          dp2[j][k] -= M;
        }
        dp2[j + 1][k + A[i]] += dp[j][k];
      }
    }
    T += A[i];
    swap(dp, dp2);
  }
  for (int i = 1; i <= S; i++){
    cout << dp[C][i];
    if (i < S){
      cout << ' ';
    }
  }
  cout << endl;
}
0