【LeetCode】Day203-最少侧跳次数

题目

1824. 最少侧跳次数【中等】

题解

这种后面问题和前面关联的题就容易想到动态规划，问题是dp怎么定义。

用0,1,2对跑道进行重新编号，obstacles[i]-1与跑道对应，-1代表无障碍物。

1. 状态定义：dp[i][j] 表示青蛙到达 第 i 点 的 第 j 条 跑道所需要的最小侧跳次数。
2. 状态转移方程：
   怎么到达的当前位置呢？有两种方法，第一是从i-1点横着跳过来，第二是从i点其他跑道侧跳过来。
   首先，如果当前跑道无障碍，青蛙可以不侧跳，直接从i-1点跳到i点，dp[i][j]=dp[i-1][j]；如果有障碍，设置dp[i][j]=inf
   然后，青蛙可以从i点其他跑道侧跳到当前跑道，设minCnt=min(d[i][0],d[i][1],d[i][2])，此时d[i][j]=min(d[i][j],minCnt+1)，其中j≠obstacles[i]−1。
3. 初始条件：
   青蛙最开始所在的1号跑道，dp[0][1]=0，此时可以通过侧跳一次进入其他跑道，因此dp[0][0]=1，dp[0][2]=1
   其余状态均赋无穷大值。
4. 返回值：min( dp[n][0],dp[n][1],dp[n][2] )

除此之外，由于dp[i]求解时只用到了dp[i-1]，因此可以利用滚动数组优化空间复杂度。

class Solution {
    static final int INF=0xfffffff;
    public int minSideJumps(int[] obstacles) {
        int n=obstacles.length-1;
        int[] dp=new int[3];
        Arrays.fill(dp,1);
        dp[1]=0;
        for(int i=1;i<=n;i++){
            int min=INF;
            //从i-1点横着跳过来
            for(int j=0;j<3;j++){
                //有障碍
                if(j==obstacles[i]-1)
                    dp[j]=INF;
                //无障碍，从i-1点跳到i点
                else
                    min=Math.min(min,dp[j]);//min是dp[0],dp[1],dp[2]中最小的
            }
            //从i点其他跑道侧跳过来
            for(int j=0;j<3;j++){
                if(j==obstacles[i]-1)
                    continue;
                dp[j]=Math.min(dp[j],min+1);//用min更新dp
            }
        }
        return Math.min(Math.min(dp[0],dp[1]),dp[2]);
    }
}

时间复杂度：$O(n)$

空间复杂度：$O(1)$

p.s 大家除夕快乐哦！！！别忘了吃饺子