poj 3278 BFS

  poj3278

  都說水題，都說是自己第一道BFS題，哈哈。通過這題，又對BFS的基本框架有了進一步的認識。

 分三個方向來BFS，主要時還是判斷邊界問題。

#include <iostream>
#include <fstream>
#include <cstdio>
#include <queue>
#include <memory.h>

#define MAX 100005
using namespace std;

queue<int> q;
int step[MAX]; //跟隨陣列，記錄步數
bool visit[MAX];
int n,k;
int next,head;

int bfs()
{
    q.push(n);
    step[n]=0;
    visit[n]=1;
while(!q.empty())
    {
        head=q.front();
        q.pop();
//分三個方向BFS
        for(int i=0; i<3 ;i++)
        {
if(i==0) next=head-1;
else if(i==1) next=head+1;
else  next=head*2;
if(next>MAX || next<0 )  continue;  //必須先判斷，不然會RE，陣列越界了
            if(!visit[next])
            {
              q.push(next);
              step[next]=step[head]+1;
              visit[next]=1;
            }
if(next==k)  return step[next];
        }
    }
return -1;
}

int main()
{
    memset(visit,0,sizeof(visit));
    scanf("%ld%ld",&n,&k);
if(n>=k)
    {
        printf("%d",n-k);
    }
else
    {
       printf("%d",bfs());
    }
return 0;
}