【DP基础】核电站问题

题目描述

一个核电站有N个放核物质的坑，坑排列在一条直线上。如果连续M个坑中放入核物质，则会发生爆炸，于是，在某些坑中可能不放核物质。 现在，请你计算：对于给定的N和M，求不发生爆炸的放置核物质的方案总数(n <= 50, m <= 5)

输入

输入文件只有一行，两个正整数N,M。

输出

输出文件只有一个正整数，表示方案总数。

样例输入

4 3 

样例输出

13

我们设状态dp[i]，表示放到第i个坑的方案数，

（i<M）dp[i]=2*dp[i-1] 对于第i个位置，因为连续一段一定不超过m个坑已经填了

（i==M）dp[i]=2*dp[i-1]-1 现在可能存在连续一段 已经 填了 m个坑的情况，但只有一种方案，删掉就行了

（i>M）dp[i]=2*dp[i-1]-dp[i-m-1] 连续一段填了m个的情况有（ i-m 前的方案数 X i-m+1~i 的方案数）dp[i-m-1] 种

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

#define MAXN 50
#define MAXM 5
#define INF 0x3f3f3f3f
typedef long long int LL;

LL dp[MAXN+10];
int N,M;

int main()
{
    scanf("%d%d",&N,&M);
    int i,j;

    dp[0]=1;
    for(i=1;i<=N;++i)
    {
        if(i<M)dp[i]=2*dp[i-1];//放不放任意
        else if(i==M)dp[i]=2*dp[i-1]-1;//不能有连续的一段
        else dp[i]=2*dp[i-1]-dp[i-M-1];//i-m处一定不能放
    }

    printf("%lld\n",dp[N]);
}