Question

我无法在函数中实现回溯，我觉得我只需要一点点推动。我从代码争夺中得到一个问题，我将链接here

你需要爬上一个有n个台阶的楼梯，你决定跳上台阶进行一些额外的锻炼。您可以在一次跳跃中覆盖最多k个步骤。返回你可以采取的所有可能的跳跃序列，以爬上楼梯，整理好。

对于n = 4且k = 2，输出应为：

climbingStaircase(n, k) = [[1, 1, 1, 1], [1, 1, 2], [1, 2, 1], [2, 1, 1], [2, 2]]

我应该以一种回溯思维来解决这个问题，但回溯对我来说是新的，我很难在一个函数中实现它。我觉得我处于边缘，但只需要一点点推动。那里有没有人可以帮我解决这个问题并完全理解回溯？

Answer 1

我遇到了同样的问题，但我解决了。这是Java中的解决方案：

    ArrayList<int[]> solutions = new ArrayList<int[]>();

    int[][] climbingStaircase(int n, int k) {
        climb(new int[n*k], 0, 0, 0, n, k);
        return trimSolution(solutions);
    }

    void climb(int[] sol, int index, int step, int sum, int max, int stepMax) {
        if(step != 0) {
            sum += step;
            sol[index++] = step;
            if(sum < max) {
                // printArray("it: ", sol);
            } else if(sum == max){
                // printArray("sol: ", sol);
                solutions.add(trimSolution(sol));
                sol[--index] = 0;
                return;
            } else {
                sol[--index] = 0;
                // printArray("failed: ", sol);
                return;
            }
        }

        for (step = 1; step <= stepMax; step++) {
            // System.out.println("index: " + index);
            climb(sol, index, step, sum, max, stepMax);
        }
    }

    int[] trimSolution(int[] sol) {
        int length = 0;
        for (int i = 0; i < sol.length; i++) {
            if(sol[i] != 0)
                length++;
        }
        int[] r = new int[length];
        for (int i = 0; i < r.length; i++) {
            r[i] = sol[i];
        }
        return r;
    }

    int[][] trimSolution(ArrayList<int[]> sol) {
        if(sol.size() == 0)
            sol.add(new int[0]);
        int[][] r = new int[sol.size()][1];
        for (int i = 0; i < sol.size(); i++) {
            r[i] = sol.get(i);
        }
        return r;
    }

void printArray(String message, int[] a) {
    System.out.print(message);
    for (int i = 0; i < a.length; i++) {
        System.out.print(a[i] + ", ");
    }
    System.out.println();
}

有人可以帮我实现JS中爬楼梯练习的回溯吗？

1 个答案: