机器人走路问题优化解法

复制代码
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 分钟前
二刷 苍穹外卖day10(含bug修改)
java·开发语言
DKPT5 分钟前
Java设计模式之结构型模式(外观模式)介绍与说明
java·开发语言·笔记·学习·设计模式
缘来是庄8 分钟前
设计模式之外观模式
java·设计模式·外观模式
2401_8812444024 分钟前
Treap树
数据结构·算法
乌萨奇也要立志学C++25 分钟前
二叉树OJ题(单值树、相同树、找子树、构建和遍历)
数据结构·算法
网安INF27 分钟前
深度学习中的逻辑回归:从原理到Python实现
人工智能·python·深度学习·算法·逻辑回归
wsxqaz39 分钟前
浏览器原生控件上传PDF导致hash值不同
算法·pdf·哈希算法
知其然亦知其所以然1 小时前
JVM社招面试题:队列和栈是什么?有什么区别?我在面试现场讲了个故事…
java·后端·面试
harmful_sheep1 小时前
Spring 为何需要三级缓存解决循环依赖,而不是二级缓存
java·spring·缓存
星辰大海的精灵1 小时前
如何确保全球数据管道中的跨时区数据完整性和一致性
java·后端·架构