POJ3278 Catch that cow

Description

在一个数轴上开始农夫在位置N，奶牛在位置K，奶牛的位置不动，农夫每次有三种选择：

1，向左走一步，花费1秒 2，向右走一步，花费一秒 3，走到2*n的位置，花费一秒

问农夫最少花多长时间找到奶牛。

Input

N和K

Output

最短的时间

题解：是一道bfs的入门题，注意只有向左走一步才能往左走，注意添加if(k<n)printf("%d ",n-k)；的剪枝。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define maxn 1000005

using namespace std;

int step[maxn],n,k;
bool vis[maxn];
queue <int> q;

inline void bfs(int pos)
{
    step[pos]=0;
    vis[pos]=true;
    q.push(pos);
    while(!q.empty())
    {
        int p=q.front(); q.pop();
        for(register int i=1;i<=3;++i)
        {
            int nxt;
            if(i==1) nxt=p+1;
            else if(i==2) nxt=p-1;
            else if(i==3) nxt=p*2;
            if(nxt<0||nxt>=maxn) continue;
            if(!vis[nxt])
            {
                step[nxt]=step[p]+1;
                vis[nxt]=true;
                q.push(nxt);
            }
            if(nxt==k) return;
        }
    }
    return;
}

int main()
{
    while(!q.empty()) q.pop();
    scanf("%d%d",&n,&k);
    if(n>=k)
    {
        printf("%d
",n-k);
        return 0;
    }
    bfs(n);
    printf("%d
",step[k]);
    return 0;
}