字节跳动笔试1——马里奥弹跳板游戏(BFS)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u012525096/article/details/94412708

题目

一个数组代表跳板,数字代表向前或向后可以跳的最大距离(比如3,那么跳1,、2、3都行),0代表悬崖(跳上去摔死)。
指定一个出生点P,求到终点需要跳的最小次数(终点是最后一个元素后面),无法到达输出-1

输入:
第一行:数组长度N,出生点P
第二行:N个数字
输出:
最小次数

示例:

输入:
7 4
10 0 2 1 1 0 1

输出:
3

1<= N <= 10000
1 <= P <= N

思路

LeetCode的jump game II, 但他只能往后跳。

最小次数,那么就是BFS了,动手写代码

代码

用一个队列实现BFS。

package shen.leetcode.solution.toutiao;

import java.util.LinkedList;

public class Solution_toutiao_0630 {

    public static void main(String[] args) {
        System.out.println(core(new int[]{10, 0, 2, 1, 1, 0, 1}, 7));
    }

    public static LinkedList<Node> q = new LinkedList<>();

    public static int core(int arr[], int start) {
        q.offer(new Node(start - 1, 0));
        while (!q.isEmpty()) {
            Node curNode = q.poll();
            if (arr[curNode.index] == 0) {
                continue;
            }
            int maxLength = arr[curNode.index];
            if (curNode.index + maxLength > arr.length - 1) {// 大于长度,说明可以到达终点,返回
                return curNode.step + 1;
            }
            if (curNode.index - maxLength >= 0) { // 可以往回跳
                for (int i = curNode.index - maxLength; i <= curNode.index - maxLength + 2 * arr[curNode.index]; i++) {
                    if (i == curNode.index) continue;
                    q.offer(new Node(i, curNode.step + 1));
                }
            } else {
                int begin = maxLength - curNode.index;
                for (int i = begin; i <= begin + 2 * arr[curNode.index] - begin; i++) {
                    if (i == curNode.index) continue;
                    q.offer(new Node(i, curNode.step + 1));
                }
            }

        }
        return -1;
    }

    public static class Node {
        int index;
        int step;

        public Node(int index, int step) {
            this.index = index;
            this.step = step;
        }
    }

}

展开阅读全文

没有更多推荐了,返回首页