核电站[解题报告]

核电站

[时间限制: 1 Sec  内存限制: 128 MB]

题目描述

      一个核电站有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; 
}