Leetcode 3429. Paint House IV

• Leetcode 3429. Paint House IV
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3429. Paint House IV

1. 解题思路

这一题解法上就是一个动态规划的思路，由于题目有两个限制条件，即相邻不可以同色，以及前后同位置不可以同色，因此我们在迭代的时候需要从首尾同步考虑两两的配色组合，剩下的就是简单的动态规划遍历所有可能性即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def minCost(self, n: int, cost: List[List[int]]) -> int:
        
        @lru_cache(None)
        def dp(idx, prefix, suffix):
            if idx == n // 2:
                return 0
            i, j = idx, n-1-idx
            ans = math.inf
            for c1 in range(3):
                for c2 in range(3):
                    if c1 == c2 or c1 == prefix or c2 == suffix:
                        continue
                    ans = min(ans, cost[i][c1] + cost[j][c2] + dp(idx+1, c1, c2))
            return ans
        
        return dp(0, -1, -1)

提交代码评测得到：耗时2108ms，占用内存143MB。