【BFS】抓住那头牛

poj 4001 抓住那头牛

抓住那头牛

农夫知道一头牛的位置，想要抓住它。农夫和牛都位于数轴上，农夫起始位于点N(0<=N<=100000)，牛位于点K(0<=K<=100000)。农夫有两种移动方式：

1、从X移动到X-1或X+1，每次移动花费一分钟
2、从X移动到2*X，每次移动花费一分钟

假设牛没有意识到农夫的行动，站在原地不动。农夫最少要花多少时间才能抓住牛？

输入
两个整数，N和K
输出
一个整数，农夫抓到牛所要花费的最小分钟数
样例输入

5 17

样例输出

4

思路

• BFS
• 可能BFS大多应用在方格图上，而这道题是一维的，不过也不影响罢了
• BFS典型模板题，直接套BFS模板即可

代码如下

#include <iostream>
#include <queue>
#include <set>
using namespace std;

//农夫走的情况
int get_to(int x, int p) {
    switch (p) {
        case 2:
            return x - 1;
        case 1:
            return x + 1;
        case 0:
            return 2 * x;
        default:
            return 0x3f3f3f3f;//形式上的一个整数，随便是多少，但是要返回一个
    }
}

void solve() {
    int n = 0, m = 0;
    queue<pair<int, int> > q;
    set<int> st;
    pair<int, int> top;
    int x = 0, i = 0;

    scanf("%d%d", &n, &m);

    //根据题目情况，当农夫的位置在牛的右边时，肯定不可能走x+1和2x的情况，所以这种情况可以剪枝
    if (n >= m) {
        printf("%d\n", n - m);
        return;
    }

    //BFS模板
    q.push(make_pair(n, 0));
    st.insert(n);
    while (!q.empty()) {
        top = q.front();
        q.pop();
        for (i = 0; i < 3; i++) {
            x = get_to(top.first, i);
            //剪枝：边界情况，是否被访问等，边界不控制会MLE
            if (x < 0 || x > 100000 || st.find(x) != st.end()) {
                continue;
            }
            if (x == m) {
                printf("%d\n", top.second + 1);
                return;
            }
            st.insert(x);
            q.push(make_pair(x, top.second + 1));
        }
    }
}

int main(void) {
	solve();
    return 0;
}