核电站
[时间限制: 1 Sec 内存限制: 128 MB]题目描述
一个核电站有N个放核物质的坑,坑排列在一条直线上。如果连续M个坑中放入核物质,则会发生爆炸,于是,在某些坑中可能不放核物质。
现在,请你计算:对于给定的N和M,求不发生爆炸的放置核物质的方案总数。
现在,请你计算:对于给定的N和M,求不发生爆炸的放置核物质的方案总数。
输入
输入只有一行,两个正整数N,M。
输出
输出只有一个正整数,表示方案总数。
样例输入
4 3
样例输出
13
思路
首先,这是一道显而易见的动规问题。
根据题意,每个坑有两种状态,即为装,或不装。
因为只有两种方案所以f[i]既是表示装了i的方案数,也是不装i的方案数。
那么,a[i]装的是第几个,有3种情况
1.第<m个, f[i]=2*f[i-1];
2.第=m个,f[i]=2*f[i-1]-1;
3.第>m个,f[i]=f[i-1]+f[i-1]-f[i-m-1];
代码
#include<bits/stdc++.h>
using namespace std;
long long f[6666666];int n,m,i;
int main()
{
cin>>n>>m;
f[0]=1;
for(i=1;i<=n;i++)
{
if(i<m)f[i]=2*f[i-1];
if(i==m)f[i]=2*f[i-1]-1;
if(i>m)f[i]=f[i-1]+f[i-1]-f[i-m-1];
}
cout<<f[n]<<endl;
return 0;
}