結果

問題 No.822 Bitwise AND
ユーザー ngtkana
提出日時 2020-04-05 16:10:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 1,066 bytes
コンパイル時間 2,553 ms
コンパイル使用メモリ 196,496 KB
最終ジャッジ日時 2025-01-09 14:15:04
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#define ALL(v) std::begin(v),std::end(v)
using lint=long long;
using lubl=long double;
int main(){
    std::cin.tie(nullptr);std::ios_base::sync_with_stdio(false);
    std::cout.setf(std::ios_base::fixed);std::cout.precision(15);
    lint n,K;std::cin>>n>>K;
    if(n==0||K==0){
        std::cout<<(K==0?"1":"INF")<<'\n';
        return 0;
    }
    lint lg=std::log2(n);
    lint N=1ll<<lg+1;
    std::vector<lint>dp(N/2,0);
    dp.at(0)=true;
    for(lint i=0;i<lg;i++){
        if(n>>i&1)continue;
        lint I=1ll<<i;
        std::vector<lint>swp=dp;
        for(lint j=0;j<N/2;j++){
            lint x=dp.at(j);
            swp.at(std::abs(j-I))+=x;
            if(j+I<N/2)swp.at(j+I)+=x;
        }
        swp.swap(dp);
    }
    for(lint i=N;i<=K+N;i++){
        lint j=std::abs(i-K);
        if(j<N/2&&dp.at(j)){
            std::cout<<"INF"<<'\n';
            return 0;
        }
    }
    std::for_each(dp.begin()+1,dp.end(),[](lint&x){x/=2;});
    std::cout<<std::accumulate(dp.begin(),dp.begin()+std::min(N/2,K+1),0ll)<<'\n';
}
0