动态规划-粉刷房子

题目描述

假如有一排房子,共 n 个,每个房子可以被粉刷成红色、蓝色或者绿色这三种颜色中的一种,你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。 当然,因为市场上不同颜色油漆的价格不同,所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 3 的正整数矩阵 costs 来表示的。

例如,costs[0][0] 表示第 0 号房子粉刷成红色的成本花费;costs[1][2] 表示第 1 号房子粉刷成绿色的花费,以此类推。 请计算出粉刷完所有房子最少的花费成本。

动态规划原理

动态规划(Dynamic Programming, DP)是一种通过将原问题分解为相对简单的子问题的方式求解复杂问题的方法。这些子问题通常是原问题的较小实例,并且它们的解被保存起来以便后续使用,从而避免了重复计算。动态规划通常用于求解具有重叠子问题和最优子结构性质的问题。

解题步骤

  1. 定义状态
    • 在这个问题中,我们定义 dp[i][j] 为将前 i 个房子粉刷完毕,并且第 i 个房子粉刷成颜色 j 的最小成本。
  2. 状态转移方程
    • 要计算 dp[i][j],我们需要考虑前一个房子 i-1 的颜色。由于相邻房子颜色不能相同,因此我们需要从 dp[i-1][k](其中 k 不等于 j)中选择最小值,并加上当前房子粉刷成颜色 j 的成本 costs[i][j]
    • 状态转移方程为:dp[i][j] = min(dp[i-1][k] + costs[i][j]),其中 k 遍历 0, 1, 2 且 k != j
  3. 初始化
    • 对于第一个房子(i = 0),我们可以直接选择任意颜色,因此 dp[0][j] = costs[0][j],其中 j = 0, 1, 2
  4. 计算顺序
    • 我们需要按照房子的顺序(从第一个房子到最后一个房子)来填充 dp 数组。
  5. 结果
    • 最终答案将是 dp[n-1][j] 中的最小值,其中 j = 0, 1, 2,表示粉刷完所有房子后的最小成本。

 代码如下:

class Solution {
public:
    int minCost(vector<vector<int>>& costs) {
        int n = costs.size();
        vector<vector<int>> dp(n, vector<int>(3, 0));
        // 初始化第一个房子的成本
        for (int i = 0; i < 3; i++) {
            dp[0][i] = costs[0][i];
        }
        // 填充dp数组
        for (int i = 1; i < n; i++) {
            for (int j = 0; j < 3; j++) {
                if (j == 0)
                    dp[i][j] =
                        min(dp[i - 1][j + 1], dp[i - 1][j + 2]) + costs[i][j];
                else if (j == 1)
                    dp[i][j] =
                        min(dp[i - 1][j - 1], dp[i - 1][j + 1]) + costs[i][j];
                else if (j == 2)
                    dp[i][j] =
                        min(dp[i - 1][j - 2], dp[i - 1][j - 1]) + costs[i][j];
            }
        }
        // 返回最小成本
        return min(dp[n - 1][0], min(dp[n - 1][1], dp[n - 1][2]));
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值