yukicoder contest 3(No.3)解説

yukicoder

No.3 ビットすごろく

No.3 ビットすごろく – yukicoder

各マスの進める距離に特徴がありますが、BFSやDFS等で最短距離を更新する全探索を行えばOKです。
“2進数で表現した時の1のビット数”ですが、2進数の下位1bitは%2で取得でき、/2をすることで右シフトさせることができます。なので、nの1のビット数が知りたい場合、nが0になるまで以下のことを繰り返せばよいです。

  1. %2が1ならカウントする
  2. /2して右シフトする
#include<bits/stdc++.h>
//#include<atcoder/all>

using namespace std;
//using namespace atcoder;

int main(){
    int N;
    cin>>N;

    int bit[N+1];
    for(int i=0;i<=N;i++){
        int n=i;
        int c=0;
        while(n!=0){
            if(n%2==1) c++;
            n/=2;
        }
        bit[i]=c;
    }

    int dp[N+1];
    for(int i=0;i<=N;i++) dp[i]=INT_MAX;

    queue<int> q;
    q.push(1);
    dp[1]=1;

    while(!q.empty()){
        int n=q.front();
        //cout<<n<<endl;
        q.pop();
        if(n-bit[n]>=1&&dp[n-bit[n]]>dp[n]+1) {
            dp[n-bit[n]]=dp[n]+1;
            q.push(n-bit[n]);
        }
        if(n+bit[n]<=N&&dp[n+bit[n]]>dp[n]+1) {
            dp[n+bit[n]]=dp[n]+1;
            q.push(n+bit[n]);
        }
    }

    if(dp[N]==INT_MAX) cout<<"-1"<<endl;
    else cout<<dp[N]<<endl;
}

コメント

タイトルとURLをコピーしました