結果

問題 No.1710 Minimum OR is X
コンテスト
ユーザー SSRS
提出日時 2021-10-15 21:50:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 1,372 bytes
コンパイル時間 1,493 ms
コンパイル使用メモリ 173,812 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 17:25:28
合計ジャッジ時間 2,517 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;
int main(){
  int N, M, K;
  cin >> N >> M >> K;
  string X;
  cin >> X;
  vector<long long> POW(N + 1);
  POW[0] = 1;
  for (int i = 0; i < N; i++){
    POW[i + 1] = POW[i] * 2 % MOD;
  }
  vector<vector<long long>> binom(N + 1, vector<long long>(N + 1, 0));
  for (int i = 0; i <= N; i++){
    binom[i][0] = 1;
    binom[i][i] = 1;
    for (int j = 1; j < i; j++){
      binom[i][j] = (binom[i - 1][j - 1] + binom[i - 1][j]) % MOD;
    }
  }
  vector<vector<long long>> dp(M + 1, vector<long long>(N + 1, 0));
  dp[0][N] = 1;
  for (int i = 0; i < M; i++){
    for (int j = 0; j <= N; j++){
      if (X[i] == '0'){
        for (int k = 0; k <= j; k++){
          long long add = dp[i][j];
          add *= POW[N - j];
          add %= MOD;
          add *= binom[j][k];
          add %= MOD;
          dp[i + 1][k] += add;
          dp[i + 1][k] %= MOD;
        }
      }
      if (X[i] == '1'){
        for (int k = 0; k <= min(j, K - 1); k++){
          long long add = dp[i][j];
          add *= POW[N - j];
          add %= MOD;
          add *= binom[j][k];
          add %= MOD;
          dp[i + 1][j] += add;
          dp[i + 1][j] %= MOD;
        }
      }
    }
  }
  long long ans = 0;
  for (int i = K; i <= N; i++){
    ans += dp[M][i];
  }
  ans %= MOD;
  cout << ans << endl;
}
0