codevs2618 核电站问题——记忆化搜索,或者dp

2618 核电站问题
 
一个核电站有N个放核物质的坑,坑排列在一条直线上。如果连续M个坑中放入核物质,则会发生爆炸,于是,在某些坑中可能不放核物质。

任务:对于给定的N和M,求不发生爆炸的放置核物质的方案总数。

输入描述 Input Description
输入文件只一行,两个正整数N,M( 1<N<50,2≤M≤5)

输出描述 Output Description
输出文件只有一个正整数S,表示方案总数。

样例输入 Sample Input
4 3

样例输出 Sample Output
13

数据范围及提示 Data Size & Hint

( 1<N<50,2≤M≤5)

转载自:

http://blog.csdn.net/u013174702/article/details/43953177

( 1<N<50,2≤M≤5)

由于本题的时间限制,采用一般的深度优先搜索会严重超时,本人已经试过,而且由于1<N<50,

所以数据要用到64位整型。

我们这题的思路是采用记忆化搜索来解决这个问题,记忆化搜索有点动态规划的意思,

是介于动态规划和深度优先搜索的一种方法。

我们先建立一个长整形的二维数组 f[i][j], 代表代表第 i 个坑前有 j 个连续,

由题我们知道 f[i][m] = 0,  f[n][j] = 1。//?还是不明白。

说到这里就可以写程序了。

#include<stdio.h>  
#include<string.h>  
long long f[50][10];    //f[i][j]代表第i个坑前有j个连续  
int m, n;  
long long dfs(int i, int j){  
    if (j == m)  
        return 0;  
    if (i == n)  
        return 1;  
    if (f[i][j] != -1)  
        return f[i][j];  
    long long t = 0;  
    t += dfs(i + 1, j + 1);     //选中这个数  
    t += dfs(i + 1, 0);         //未选这个数  
    return f[i][j] = t;  
}  
  
int main(){  
    scanf("%d%d", &n, &m);  
    memset(f, -1, sizeof(f));  
    dfs(0, 0);  
    printf("%lld\n", f[0][0]);  
    return 0;  
}  

解法2 dp

设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;
    }



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值