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

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

一 原问题链接

3278 -- Catch That Cow

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

二 输入和输出

1 输入

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

2 输出

农夫抓牛的最小步数

三 输入和输出样例

1 输入样例

5 17

2 输出样例

4

四 代码

  1. package graph.poj3278;
  2. import java.util.LinkedList;
  3. import java.util.Queue;
  4. import java.util.Scanner;
  5. public class POJ3278BFS {
  6. static final int MAXN = 100009;
  7. static boolean vis[] = new boolean[MAXN];
  8. static int d[] = new int[MAXN];
  9. static int n, k;
  10. public static void main(String[] args) {
  11. Scanner scanner = new Scanner(System.in);
  12. n = scanner.nextInt();
  13. k = scanner.nextInt();
  14. if (k <= n) {
  15. System.out.println(n - k);
  16. return;
  17. }
  18. solve();
  19. }
  20. static void solve() {
  21. Queue<Integer> q = new LinkedList<>();
  22. vis[n] = true;
  23. d[n] = 0;
  24. q.add(n);
  25. while (!q.isEmpty()) {
  26. int u = q.peek();
  27. q.poll();
  28. if (u == k) {
  29. System.out.println(d[k]);
  30. return;
  31. }
  32. int x;
  33. x = u + 1;
  34. if (x >= 0 && x <= 100000 && !vis[x]) { // 向前走一步
  35. d[x] = d[u] + 1;
  36. vis[x] = true;
  37. q.add(x);
  38. }
  39. x = u - 1;
  40. if (x >= 0 && x <= 100000 && !vis[x]) { // 向后走一步
  41. d[x] = d[u] + 1;
  42. vis[x] = true;
  43. q.add(x);
  44. }
  45. x = u * 2;
  46. if (x >= 0 && x <= 100000 && !vis[x]) { // 跳着走
  47. d[x] = d[u] + 1;
  48. vis[x] = true;
  49. q.add(x);
  50. }
  51. }
  52. }
  53. }

五 测试

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

相关文章