题目描述
一个核电站有N个放核物质的坑,坑排列在一条直线上。如果连续M个坑中放入核物质,则会发生爆炸,于是,在某些坑中可能不放核物质。
任务:对于给定的N和M,求不发生爆炸的放置核物质的方案总数
输入格式
输入文件只一行,两个正整数N,M( 1<N<50,2≤M≤5)
输出格式
输出文件只有一个正整数S,表示方案总数。
题解:
一开始以为可以用数学方法解答,但WA了,看了别人的题解后有感。
设f[i]为第i个坑道使不爆炸的方案数。
当i<m时,f[i]=2*f[i-1];//因为第i个坑道只有放于不放2种情况
当i=m时,f[i]=2*f[i-1]-1;//减去的1是前m坑道都放的情况
当i>m时,f[i]=2*f[i-1]-f[i-m-1];//这是最难理解的部分,因为无论i-m-1的情况是什么,它后面都可以接i-m到m都是放的情况。或者说:当到第I个坑时,则第一个开始连着的坑其实是I-M+1,但是由于f[I-m]包括了第I-m个坑是“填”的情况,而这种情况下,已经出现了M+1个坑是连着“填”的情况,根据最优子结构,这种情况已经排除,因此我们只需要减去I-m个坑是“不填的情况”即f[I-m-1]
代码实现:
#include<iostream> using namespace std; int main() { long long f[50];int i,n,m; 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]=2*f[i-1]-f[i-m-1]; } cout<<f[n]<<endl; return 0; }