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