結果
問題 | No.2060 AND Sequence |
ユーザー |
![]() |
提出日時 | 2022-11-01 17:09:12 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,378 bytes |
コンパイル時間 | 4,448 ms |
コンパイル使用メモリ | 257,316 KB |
最終ジャッジ日時 | 2025-02-08 16:37:08 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 WA * 1 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>using namespace std;using namespace atcoder;using mint = modint998244353;int main() {int N, M, K = 30;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;}