摆花

所有人都是dp 来一个记忆化dfs

先来无记忆化的裸dfs

只有三十分

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
int num[105];  //num[i]: 第 i 种花最多能放多少盆
int n,m,sum;   //sum记录有多少种
inline void dfs(int x,int v){   //x: 已经搜索到第 x 种花  v:当前已经摆了 v 盆花 
     if(v > m)return;    //小小的剪枝(当前已经摆的花多余了可以最多摆的花(v > m) 别搜了返回吧.... )
     if(x == n+1){if (v == m)sum++;sum=sum%1000007;return;}  //如果枚举完所有种类的花 就到了边界  判断此时摆的盆数是否等于m 等于的话sum++
     for(int i=0;i <= num[x];i++) dfs(x+1,v+i);   //枚举第 x 种花的摆法 从0到最大 (0 <= i <= num[x])
     return;  //搜完了 回去吧
     }
int main(){
    scanf("%d%d",&n,&m);    //输入 n m
    for(register int i=1;i <= n;i++) scanf("%d",&num[i]);  // 输入每种花最多能摆多少 并储存在num里
    dfs(1,0);   //COME ON ! 去搜索!
    printf("%d",sum);  //输出sum
    return 0;  // over......
}

  

其实可以看到呀 dfs的参数只有 x v 可以证明 对于决定的x v 有唯一的 摆放的种数 于是就开一个二维数组 mem[][] 储存每种 x v 所对应的答案

代码如下:

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
long long num[105];
long long mem[105][105];
bool tf[105][105];
long long n,m;
long long dfs(long long x,long long v){
       if(v < 0) return 0;
       if(x == n) {if(v >= 0&&v <= num[x])return 1;else return 0;}
       if(tf[x][v]) return mem[x][v];
       long long countt=0;
       for(int i=0;i <= num[x];i++) {countt+=dfs(x+1,v-i);countt=countt%1000007;}
       mem[x][v]=countt;
       tf[x][v]=true;
       return mem[x][v];
     }
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i <= n;i++) scanf("%lld",&num[i]);
    printf("%lld",dfs(1,m)%1000007);
    return 0;
}

  

posted @ 2019-03-24 14:02  DATA-P  阅读(268)  评论(0)    收藏  举报