【dp】leetcode 276. Paint Fence

原创于 2017-10-06 23:59:31 发布 · 545 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

157 篇文章

订阅专栏

本文提供了一个针对LeetCode上第276题“涂漆栅栏”(Paint Fence)的高效解决方案。通过使用动态规划方法，详细介绍了如何计算在特定条件下对栅栏进行涂色的不同方式的数量。该方法考虑了相邻栅栏不能使用相同颜色的情况。

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

【dp】leetcode 276. Paint Fence

class Solution {
    public int numWays(int n, int k) {
        if(n==0||k==0) return 0;
        int[][][] dp = new int[n+1][k][2];
        for(int j=0;j<k;j++){
            dp[1][j][0] = 1;
        }
        for(int i=2;i<=n;i++){
            for(int j=0;j<k;j++){
                // dp[i][j][0]
                for(int h=1;h<k;h++){
                    dp[i][j][0] += dp[i-1][(j+h)%k][0]+dp[i-1][(j+h)%k][1];
                }
                dp[i][j][1] = dp[i-1][j][0];
            }
        }
        int sum = 0;
        for(int j=0;j<k;j++){
            for(int h=0;h<2;h++){
                sum += dp[n][j][h];
            }
        }
        return sum;
    }
}