力扣（LeetCode）276. 栅栏涂色（2022.10.03）

最新推荐文章于 2024-04-23 10:23:23 发布

ChaoYue_miku

最新推荐文章于 2024-04-23 10:23:23 发布

阅读量581

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/ChaoYue_miku/article/details/127158091

LeetCode 专栏收录该内容

434 篇文章

订阅专栏

本文介绍了一道算法题的解决方法，该问题要求计算给定栅栏柱数量和颜色种类时的有效涂色方案总数。通过动态规划算法，文章详细展示了如何避免连续三个栅栏柱颜色相同的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

有 k 种颜色的涂料和一个包含 n 个栅栏柱的栅栏，请你按下述规则为栅栏设计涂色方案：

每个栅栏柱可以用其中一种颜色进行上色。
相邻的栅栏柱最多连续两个颜色相同。
给你两个整数 k 和 n ，返回所有有效的涂色方案数。

示例 1：
在这里插入图片描述

输入：n = 3, k = 2
输出：6
解释：所有的可能涂色方案如上图所示。注意，全涂红或者全涂绿的方案属于无效方案，因为相邻的栅栏柱最多连续两个颜色相同。

示例 2：

输入：n = 1, k = 1
输出：1

示例 3：

输入：n = 7, k = 2
输出：42

提示：

1 <= n <= 50
1 <= k <= 10⁵
题目数据保证：对于输入的 n 和 k ，其答案在范围 [0, 2³¹ - 1] 内

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/paint-fence

方法一：动态规划

C++提交内容：

class Solution {
public:
    int numWays(int n, int k) {
        if (k == 1) return n > 2 ? 0 : 1;
        if (n <= 2) return pow(k, n);
        vector<int> dp(n);
        dp[0] = k;
        dp[1] = k * k;
        for (int i = 2; i < n; ++i) {
            dp[i] = dp[i - 1] * (k - 1) + dp[i - 2] * (k - 1);
        }
        return dp[n - 1];
    }
};