机器人走路问题优化解法

复制代码
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];
    }
}
相关推荐
程序员爱钓鱼5 小时前
Go语言实战案例——进阶与部署篇:编写Makefile自动构建Go项目
后端·算法·go
canonical_entropy6 小时前
DDD本质论:从哲学到数学,再到工程实践的完整指南之实践篇
java·后端·领域驱动设计
_Power_Y6 小时前
Java面试常用算法api速刷
java·算法·面试
艾醒(AiXing-w)6 小时前
大模型面试题剖析:模型微调中冷启动与热启动的概念、阶段与实例解析
人工智能·深度学习·算法·语言模型·自然语言处理
纪莫6 小时前
技术面:Spring (事务传播机制、事务失效的原因、BeanFactory和FactoryBean的关系)
java·spring·java面试⑧股
天选之女wow6 小时前
【代码随想录算法训练营——Day32】动态规划——509.斐波那契数、70.爬楼梯、746.使用最小花费爬楼梯
算法·leetcode·动态规划
红衣小蛇妖6 小时前
LeetCode-704-二分查找
java·算法·leetcode·职场和发展
!chen6 小时前
【Spring Boot】自定义starter
java·数据库·spring boot
rongqing20196 小时前
问题记录:一个简单的字符串正则匹配算法引发的 CPU 告警
算法
koko426 小时前
天津小公司面经
java·学习·面试