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

Source

ニコニコミュニティ
問題文

問題概要

$1$ から $N$ までの整数が書かれた $N$ 個のマスがある．
$f(x)$ で整数 $x$ を $2$ 進数で書いた時の $1$ の桁の数を表すとする．
今いるマスが $k$ ならば，$k+f(k)$ のマスか $k-f(k)$ のマスに（そのマスが存在すれば）移動できる．
最初 $1$ のマスにいて，$N$ のマスまで移動するための最小ステップ数を求める問題．

解法

幅優先探索すれば良い．

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

#define ll long long
#define ull unsigned ll

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void writer(int x, char c){int s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}

int N;
int q[100010], q_st, q_size, dist[100010], bt[100010];

int main(){
  int i, j, k;

  reader(&N);
  rep(i,N+1) dist[i] = -1;
  dist[1] = 1;
  q_st = q_size = 0;
  q[q_st+q_size++] = 1;

  bt[0] = 0;
  REP(i,1,N+1) bt[i] = bt[i&(i-1)]+1;

  while(q_size){
    k = q[q_st++]; q_size--;
    if(k-bt[k] >= 1 && dist[k-bt[k]]==-1){
      dist[k-bt[k]] = dist[k] + 1;
      q[q_st+q_size++] = k-bt[k];
    }
    if(k+bt[k] <= N && dist[k+bt[k]]==-1){
      dist[k+bt[k]] = dist[k] + 1;
      q[q_st+q_size++] = k+bt[k];
    }
  }

  writer(dist[N],'\n');

  return 0;
}

Current time: 2024年04月19日23時30分28秒
Last modified: 2014年11月07日20時18分01秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン：