【Leetcode】256. Paint House

题目地址：

https://leetcode.com/problems/paint-house/

给定一个二维数组，第$i$行代表房子$i$用颜色$k$刷漆需要的费用。要求相邻的房子不能刷同样颜色的漆。问粉刷所有的房子需要的最小费用是多少。

思路是动态规划。令$f[i][j]$表示第$i$个房子用第$j$种漆刷的情况下，第$0...i$刷漆的总费用。那么有$$f[i][j]=costs[i][j]+\underset{k\ne j}{\min }f[i-1][k]$$最终结果就是$f$数组的最后一行的最小值。初始条件就是第一个房子对应的费用向量。

由于我们发现，$f[i][j]$只取决于前一行，所以为了节省空间，可以用滚动数组。代码如下：

public class Solution {
    public int minCost(int[][] costs) {
        if (costs == null || costs.length == 0 || costs[0].length == 0) {
            return 0;
        }
        
        int[][] dp = new int[2][3];
        // 初始化
        for (int i = 0; i < 3; i++) {
            dp[0][i] = costs[0][i];
        }
        
        // cur代表当前需要更新的行的坐标
        int cur = 0;
        for (int i = 1; i < costs.length; i++) {
        	// 滚动一下
            cur ^= 1;
            for (int j = 0; j < 3; j++) {
                dp[cur][j] = costs[i][j] + Math.min(dp[cur ^ 1][(j + 1) % 3], dp[cur ^ 1][(j + 2) % 3]);
            }
        }
        
        int res = Integer.MAX_VALUE;
        for (int i = 0; i < 3; i++) {
            res = Math.min(res, dp[cur][i]);
        }
        
        return res;
    }
}

时间复杂度$O(n)$，空间复杂度$O(1)$。