1051-粉刷房子

题目如下

假如有一排房子，共 n 个，每个房子可以被粉刷成红色、蓝色或者绿色这三种颜色中的一种，你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。

当然，因为市场上不同颜色油漆的价格不同，所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 3 的正整数矩阵 costs 来表示的。

例如，costs[0][0] 表示第 0 号房子粉刷成红色的成本花费；costs[1][2] 表示第 1 号房子粉刷成绿色的花费，以此类推。

请计算出粉刷完所有房子最少的花费成本。

示例 1：
输入: costs = [[17,2,17],[16,16,5],[14,3,19]]
输出: 10
解释: 将 0 号房子粉刷成蓝色，1 号房子粉刷成绿色，2 号房子粉刷成蓝色。
最少花费: 2 + 5 + 3 = 10。

示例 2：
输入: costs = [[7,6,2]]
输出: 2

解题思路

每一步需要粉刷一幢房子，粉刷所有房屋需要 n 步，每一步又面临粉刷成红绿蓝三色的选择，最终需要计算最少成本，所以此题可使用动态规划求解。

设 r(i) 、g(i) 和 b(i) 分别表示将标号为 i 的房屋刷成红色、绿色和蓝色时粉刷标号从 0 到 i 的 i + 1 幢房子的最小成本。同时因为相邻的房屋不能涂成同一个颜色，所以可以得到状态转移方程为

完整代码如下，时间复杂度为 O(n)，空间复杂度为 O(1)。

class Solution 
{
public:
    int minCost(vector<vector<int>>& costs)
    {
        vector<vector<int>>dp(2, vector<int>(3, 0));
        dp[0] = costs[0];
        for (int i = 1; i < costs.size(); ++i) 
        {
            for (int j = 0; j < 3; ++j) 
            {
                int prev1 = dp[(i - 1) % 2][(j + 1) % 3];
                int prev2 = dp[(i - 1) % 2][(j + 2) % 3];
                dp[i % 2][j] = min(prev1, prev2) + costs[i][j];
            }
        }

        int last = (costs.size() - 1) % 2;
        return min(min(dp[last][0], dp[last][1]), dp[last][2]);
    }
};

class Solution
{
public:
    int minCost(vector<vector<int>>& vec)
    {
        int[][] dp = new int[costs.length][3];
        //第一件房子
        for (int i = 0; i < 3; i++)
        {
            dp[0][i] = costs[0][i];
        }
        for (int i = 1; i < costs.length; i++)
        {
            //状态转换
            dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + costs[i][0];
            dp[i][1] = Math.min(dp[i - 1][0], dp[i - 1][2]) + costs[i][1];
            dp[i][2] = Math.min(dp[i - 1][0], dp[i - 1][1]) + costs[i][2];
        }
        return Math.min(dp[costs.length - 1][0],
            Math.min(dp[costs.length - 1][1], dp[costs.length - 1][2]));
    }
};