栅栏染色

最新推荐文章于 2021-07-29 11:19:38 发布

这题我会啊

最新推荐文章于 2021-07-29 11:19:38 发布

阅读量404

点赞数 1

分类专栏： Algorithm 文章标签：动态规划

本文链接：https://blog.csdn.net/Love_wanling/article/details/62418372

版权

Algorithm 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题目描述有误是至多两个相邻一样

此题用动态规划做因为第j个柱子分为两种情况：

和j-1一样
和j-1不一样

和j-1一样则为 f[j-2]*(k-1); 和j-1不一样则为 f[j-1]*(k-1)

第一次忘记考虑特殊情况柱子很多只给一种颜色的无解情况

class Solution:
    # @param {int} n non-negative integer, n posts
    # @param {int} k non-negative integer, k colors
    # @return {int} an integer, the total number of ways
    def numWays(self, n, k):
        # Write your code here
        if k == 1 and n > 2:
            return 0
        if n == 1:
            return k
        if n == 2:
            return k*k
        return (k-1)*(self.numWays(n-1,k)+self.numWays(n-2,k))