結果

問題 No.2060 AND Sequence
ユーザー ripity
提出日時 2022-11-01 17:10:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 284 ms / 2,000 ms
コード長 1,378 bytes
コンパイル時間 4,467 ms
コンパイル使用メモリ 256,820 KB
最終ジャッジ日時 2025-02-08 16:37:32
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
int main() {
  int N, M, K = 31;
  cin >> N >> M;
  vector<vector<mint>> dp(N, vector<mint>(K)), binom(K, vector<mint>(K, 1));
  vector<vector<vector<mint>>> ddp(K, vector<vector<mint>>(K, vector<mint>(2)));
  if( ((M>>(K-1))&1) == 0 ) {
    ddp[K-1][0][1] = 1;
  }else {
    ddp[K-1][0][0] = 1;
    ddp[K-1][1][1] = 1;
  }
  for( int i = K-2; i >= 0; i-- ) {
    int D = (M>>i)&1;
    for( int j = 0; j < K; j++ ) {
      for( int k = 0; k < 2; k++ ) {
        for( int d = 0; d < 2; d++ ) {
          if( j-d < 0 || ( k == 1 && d > D ) ) continue;
          int k2 = ( d == D );
          ddp[i][j][k&k2] += ddp[i+1][j-d][k];
        }
      }
    }
  }
  for( int n = 0; n < K; n++ ) {
    for( int r = 0; r <= n; r++ ) {
      for( int k = 1; k <= n; k++ ) binom[n][r] *= k;
      for( int k = 1; k <= r; k++ ) binom[n][r] /= k;
      for( int k = 1; k <= n-r; k++ ) binom[n][r] /= k;
    }
  }
  for( int j = 0; j < K; j++ ) {
    dp[0][j] = ddp[0][j][0]+ddp[0][j][1];
  }
  for( int i = 0; i < N-1; i++ ) {
    for( int j = 0; j < K; j++ ) {
      for( int k = 0; k <= j; k++ ) {
        dp[i+1][k] += dp[i][j]*binom[j][k];
      }
    }
  }
  mint ans = 0;
  for( int j = 0; j < K; j++ ) {
    ans += dp[N-1][j];
  }
  cout << ans.val() << endl;
}
0