結果

問題 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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0