积木问题

     有N块积木，我们需要用这些积木造塔。每个塔有H层，最底层包含M块积木；对于上面的每一层，包含的积木块数必须比下面一层的多1或者少1.

给定输入条件：积木数N，塔的层数H，最底层的积木数M，积木不一定要用完。求方案总数

 

使用动态规划的方法。同时使用剪枝条件进行剪枝

剪枝条件：当前剩余的积木数量不足以构建H层的塔，则终止当前的方案

               当前剩余的积木数量大于所需的最大数量：每层递增1的所有层的和。同时如果底层积木数M大于层数H，则有2^h 种方案。当前构造停

               止。

动态规划的方法：设当前层积木数量为c，剩余积木数为a，当前层数为b。则所需最大积木数为(h-b+1)*(c+c+h-b)/2;，所需最小积木数为(c+1)*c/2+2*(h-b+1-c)-(h-b+1-c)/2;

                     F(a,b,c)=f(a-c,b+1,c+1)+f(a-c,b+1,c-1)

 

代码如下

#include "stdafx.h" #include <iostream> #include <memory.h> #define maxn 2000 #define maxnM 20 struct fType { int n;// available blocks int h;// the total num of floors; int m;// the num of blocks used in current floor; int num;//the num of solutions; }; class BuildingBlock { fType f[maxn]; bool used[maxn]; bool did[maxnM][maxnM][maxnM]; int n,h,m; public: BuildingBlock() { memset(used,false,maxn); for(int i=0;i<maxnM;i++) { for(int j=0;j<maxnM;j++) for(int k=0;k<maxnM;k++) did[i][j][k]=false; } std::cin>>n>>h>>m; std::cout<<search(n,1,m)<<std::endl; } void insert(int a,int b,int c,int num) { int index=(a*b*c)%maxn; while(used[maxn]) { index=(index++)%maxn; } f[index].n=a; f[index].h=b; f[index].m=c; f[index].num=num; used[index]=true; did[a][b][c]=true; } int get(int a,int b,int c) { int index=(a*b*c)%maxn; while(!(f[index].n==a&&f[index].h==b&&f[index].m==c)) index=(index+1)%maxn; return f[index].num; } int search(int a,int b,int c) { if(a<=0||b<=0||c<=0) return 0; if(did[a][b][c]) return get(a,b,c); //int min=(h-b+1)*(c+c-(h-b))/2; int min=(c+1)*c/2+2*(h-b+1-c)-(h-b+1-c)/2; int max=(h-b+1)*(c+c+h-b)/2; if(a<min) return 0; if(a>=max) { if(c>=(h-b+1)) return 1<<(h-b); a=max; } int f=search(a-c,b+1,c+1)+search(a-c,b+1,c-1); insert(a,b,c,f); return f; } };