所有人都是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;
}