核电站问题
一个核电站有N个放核物质的坑,坑排列在一条直线上。如果连续M个坑中放入核物质,则会发生爆炸,
于是,在某些坑中可能不放核物质。
任务:对于给定的N和M,求不发生爆炸的放置核物质的方案总数。
输入描述 Input Description
输入文件只一行,两个正整数N,M( 2<=N<50,2≤M≤5)
输出描述 Output Description
输出文件只有一个正整数S,表示方案总数。
样例输入 Sample Input
4 3
样例输出 Sample Output
13
题解:
方法一:
从第一个坑开始放,开始dfs,每个坑都有两种可能,放或者不放,dfs(i,j)i第i个坑,j前i个坑最后j个坑都填了物质,若j达到m,则次方案不行,若没有m,继续
#include<stdio.h>
int n,m,ans=0;
int dfs(int i,int j){
if(j==m) return 0;//如果有连续的m坑都有物质,此方案不行
if(i==n) {
ans++;//能到n,说明之前没有连续的m坑都有物质,此方案可以
return 1;
}
int ans=0;
dfs(i+1,0);//第i+1个坑里没有物质,之后的坑里是否放物质与前面没有联系了
dfs(i+1,j+1);//前i+1个坑中最后连续j+1个坑里都有物质,
}
int main(){
int res;
scanf("%d%d",&n,&m);
dfs(0,0);//从第0个坑里开始放
printf("%d",ans);
}
方二:
dp[n]表示在n个坑中,每个坑都有两种可能,放或者不放,但不能有连续m个坑里都有物质
从头开始向后放,每个坑都有两种可能,放或者不放
开头前i个连续的坑中都有核物质(0<=i<=m-1),则第i+1个坑必空。 i+1后面的坑再放和前面的就又没关系了
i=0时的放法为dp[n-1]; 在后n-1个坑中,每个坑都有两种可能,放或者不放,但不能有连续m个坑里都有物质
i=1时的放法为dp[n-2]; 在后n-2个坑中…………
……………………;
i=m-1时的放法为dp[n-m]。 在后n-m个坑中…………
得dp[n]=dp[n-1]+dp[n-2]+……+dp[n-m] 。
当n<=m-1时: dp[n]=2*dp[n-1] (无论放还是不放,都不会有连续m个坑里都有)
当n==m时: dp[n]=2*dp[n-1] -1(减去每个坑里都有核物质这种情况)
当n>m时:
dp[n]=dp[n-1]+dp[n-2]+……+dp[n-m]
dp[n-1]=dp[n-2]+……+dp[n-m]
两式相减得:
dp[n]=dp[n-1]*2-dp[n-m-1]
#include<stdio.h>
int main(){
int n,m,i;
scanf("%d%d",&n,&m);
long long dp[n+1];
dp[0]=1;//n>1,m>1;
for(i=1;i<=n;i++){
if(i<m) dp[i]=dp[i-1]*2;
else if(i==m) dp[i]=dp[i-1]*2-1;
else dp[i]=dp[i-1]*2-dp[i-m-1];
}
printf("%lld",dp[n]);
}