結果

問題 No.1531 く2
ユーザー chocorusk
提出日時 2021-06-05 00:25:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 1,658 bytes
コンパイル時間 6,864 ms
コンパイル使用メモリ 181,512 KB
最終ジャッジ日時 2025-01-22 03:24:12
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
#define popcountll __builtin_popcountll
using mint=modint998244353;
int main()
{
    ll n, k; cin>>n>>k;
    mint ans=0;
    mint dp[2][62][62];
    dp[0][0][0]=1;
    for(int i=0; i<60; i++){
        for(int j=0; j<60; j++){
            if(k&(1ll<<i)){
                dp[0][i+1][j+1]+=dp[0][i][j];
                dp[1][i+1][j]+=dp[1][i][j];
            }else{
                dp[0][i+1][j]+=dp[0][i][j];
                dp[0][i+1][j+1]+=dp[1][i][j];
            }
            if(n&(1ll<<i)){
                if(k&(1ll<<i)){
                    dp[1][i+1][j]+=dp[0][i][j];
                    dp[1][i+1][j+1]+=dp[1][i][j];
                }else{
                    dp[0][i+1][j+1]+=dp[0][i][j];
                    dp[1][i+1][j]+=dp[1][i][j];
                }
            }
        }
    }
    for(int i=0; i<=60; i++){
        ans+=mint(2).pow(i)*dp[0][60][i];
        ans+=mint(2).pow(i+1)*dp[1][60][i];
    }
    // for(int i=0; i<=n; i++){
    //     if((n&i)==i) ans+=mint(2).pow(popcountll(k+i));
    // }
    cout<<ans.val()<<endl;
    return 0;
}
0