白色巧克力高度为1cm,黑色巧克力有两种,高度分别为1cm,kcm,把巧克力放进罐子里,要求最下面一块和最上面一块必须都是黑色,黑白相间放,给出n,问高度从1到n的罐子有共有多少种方法。
DP题目,dp[i][0]代表高度为i的罐子最后一块为白色符合其他要求的方法,dp[i][1]代表高度为i的罐子最后一块为黑色的符合所有要求的方法。
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long llu;
ll dp[105][5];//1 黑 0 白
int main()
{
int n,k;
scanf("%d%d",&n,&k);
dp[1][0] = 0;
dp[1][1] = dp[k][1] = 1;
for(int i=2;i<=n;i++)
{
dp[i][1] += dp[i - 1][0];
if(i + 1 >= k) dp[i][1] += dp[i - k][0];
dp[i][0] += dp[i - 1][1];
}
ll ans = 0;
for(int i=1;i<=n;i++) ans += dp[i][1];
printf("%lld\n",ans);
return 0;
}