机器人走路问题优化解法

复制代码
public class Test53 {
    //假设有N个位置,记为1-N,N大于或等于2
    //开始机器人在M位置上(M为1-N中的一个)
    //如果机器人来到1位置,那么下一步只能向右来到2位置
    //如果机器人来到N位置,那么下一步只能向左来到N-1的位置
    //如果机器人在中间,那么既可以往左也可以往右
    //规定机器人走K步,最终来到P位置的方法有多少种
    //给N,M,K,P,返回数量
    public static int ways2(int N, int M, int K, int P) {
        if (N < 2 || K < 1 || M < 1 || M > N || P < 1 || P > N) {
            return 0;
        }
        int[][] dp = new int[N+1][K+1];
        for (int row = 0; row <= N; row++) {
            for (int col = 0; col <= K; col++) {
                dp[row][col] = -1;
            }
        }
        return walk(N, M, K, P, dp);
    }

    public static int walk(int N, int cur, int rest, int P, int[][] dp) {
        if (dp[cur][rest] != -1) {
            return dp[cur][rest];
        }
        if (rest == 0) {
            dp[cur][rest] = cur == P ? 1 : 0;
            return dp[cur][rest];
        }
        if (cur == 1) {
            dp[cur][rest] = walk(N, 2 ,rest - 1, P, dp);
            return dp[cur][rest];
        }
        if (cur == N) {
            dp[cur][rest] = walk(N, N - 1, rest - 1, P, dp);
            return dp[cur][rest];
        }
        dp[cur][rest] = walk(N, cur + 1, rest - 1, P, dp) + walk(N, cur - 1, rest - 1, P, dp);
        return dp[cur][rest];
    }
}
相关推荐
坚持编程的菜鸟17 分钟前
LeetCode每日一题——二进制求和
c语言·算法·leetcode
lang2015092826 分钟前
Spring Boot RSocket:高性能异步通信实战
java·spring boot·后端
Glink35 分钟前
现在开始将Github作为数据库
前端·算法·github
默默coding的程序猿1 小时前
1.北京三维天地公司-实施实习生
java·sql·技术支持·面经·实施·实施工程师·三维天地
WWZZ20251 小时前
快速上手大模型:机器学习6(过拟合、正则化)
人工智能·算法·机器学习·计算机视觉·机器人·slam·具身感知
ceclar1231 小时前
C++Lambda表达式
开发语言·c++·算法
天天摸鱼的java工程师1 小时前
解释 Spring 框架中 bean 的生命周期:一个八年 Java 开发的实战视角
java·后端
尤老师FPGA1 小时前
LVDS系列32:Xilinx 7系 ADC LVDS接口参考设计(三)
android·java·ui
自由的疯1 小时前
Java 如何学习 Jenkins?
java·架构
2401_841495641 小时前
【强化学习】动态规划算法
人工智能·python·算法·动态规划·强化学习·策略迭代·价值迭代