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;
}