276. Paint Fence

There is a fence with n posts, each post can be painted with one of the k colors.

You have to paint all the posts such that no more than two adjacent fence posts have the same color. 

Return the total number of ways you can paint the fence. 

Note:
n and k are non-negative integers.

注意注意!这里是 no more than two adjacent fence posts have the same color,连续涂同一种颜色不超过两个,也就是说可以相邻的两个涂同一种颜色,第三个就不能涂同一种颜色了。所以,定义4个变量,currdiff(当前的涂不同颜色),currsame(当前的涂相同颜色),prediff(之前一个涂不同颜色),presame(之前一个涂相同颜色)。动起来的步骤是:

1、prediff = currdiff,presame = cursame

2、currdiff = (prediff + presame)* (k - 1)

3、currsame = prediff

代码实现如下:

public class Solution {
    public int numWays(int n, int k) {
        if (n == 0 || k == 0) {
            return 0;
        }
        if (n == 1) {
            return k;
        }
        int currsame = k;
        int currdiff = k * (k - 1);
        for (int i = 2; i < n; i ++) {
            int prediff = currdiff;
            int presame = currsame;
            currdiff = (prediff + presame) * (k - 1);
            currsame = prediff;
        }
        return currdiff + currsame;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值