Secret of Chocolate Poles （dp）

思路

根据题目给出的性质，显然可以用dp

dp[i][j] i表示目前高度为i层，j表示白色或者黑色，0为白，1为黑。

因为题目要求黑白交错叠，故白的都由黑的转移过来，黑的都由白的转移过来。

在层数小于k时有
dp[i][0] = dp[i - 1][1]
dp[i][1] = dp[i -1][0]

当层数大于等于k时dp[i][1]的状态转移方程多了一项
dp[i][1] = d[i - 1][0] + dp[i - k][0]

代码

#include <bits/stdc++.h>
#define _for(i,a,b) for( int i=(a); i<(b); ++i)
#define _rep(i,a,b) for( int i=(a); i<=(b); ++i)
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 10;
const int INF = 0x3f3f3f3f;

ll dp[maxn][5];
int main()
{
    int l, k;
    while(cin >> l >> k)
    {
        dp[0][0] = 1;
        dp[0][1] = 0;
        _rep(i, 1, l)
        {
            dp[i][1] = dp[i - 1][0];
            dp[i][0] = dp[i - 1][1];
            if(i >= k)
                dp[i][1] = dp[i - 1][0] + dp[i - k][0];
        }
        ll ans = 0;
        _rep(i, 1, l)
        {
            ans += dp[i][1];
        }
        cout << ans << endl;
    }
    return 0;
}