广度度优先搜索实现抓牛问题

x33g5p2x  于2022-06-29 转载在 其他  
字(1.0k)|赞(0)|评价(0)|浏览(284)

一 原问题链接

3278 -- Catch That Cow

http://poj.org/problem?id=3278

二 输入和输出

1 输入

两个数,第1个数代表农夫的位置,第2个数代表牛的位置

2 输出

农夫抓牛的最小步数

三 输入和输出样例

1 输入样例

5 17

2 输出样例

4

四 代码

package graph.poj3278;

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class POJ3278BFS {
    static final int MAXN = 100009;
    static boolean vis[] = new boolean[MAXN];
    static int d[] = new int[MAXN];
    static int n, k;

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        n = scanner.nextInt();
        k = scanner.nextInt();

        if (k <= n) {
            System.out.println(n - k);
            return;
        }
        solve();
    }

    static void solve() {
        Queue<Integer> q = new LinkedList<>();
        vis[n] = true;
        d[n] = 0;
        q.add(n);
        while (!q.isEmpty()) {
            int u = q.peek();
            q.poll();
            if (u == k) {
                System.out.println(d[k]);
                return;
            }
            int x;
            x = u + 1;
            if (x >= 0 && x <= 100000 && !vis[x]) { // 向前走一步
                d[x] = d[u] + 1;
                vis[x] = true;
                q.add(x);
            }
            x = u - 1;
            if (x >= 0 && x <= 100000 && !vis[x]) { // 向后走一步
                d[x] = d[u] + 1;
                vis[x] = true;
                q.add(x);
            }
            x = u * 2;
            if (x >= 0 && x <= 100000 && !vis[x]) { // 跳着走
                d[x] = d[u] + 1;
                vis[x] = true;
                q.add(x);
            }
        }
    }
}

五 测试

绿色为输入,白色为输出。

相关文章