P1731生日蛋糕 减枝

//这道题可以通过一般的搜索找到思路，但是很显然会超时所以需要优化

//其中需要三重优化：第一重是如果已用面积+最小面积仍然超过最优答案就返回

//第二重是如果已用体积+最小体积仍然超过要求体积就返回

//第三重剪枝是假设剩余所有的体积都用来做下一层那么此时下一层的体积是最大，而半径会最大，从而表面积最小

#include<iostream>
#include<cmath>
using namespace std;
int m,n,Minv[21],Mins[21],Best;
void Read()
{  int i;
   cin>>n>>m;
   Minv[0]=0;
   for(i=1;i<=m;i++)
   {
        Mins[i]=Mins[i-1]+2*i*i; 
        Minv[i]=Minv[i-1]+i*i*i;  //初始化最小体积和
   } 
}
void search(int floor,int r,int h,int s,int v)//对每层蛋糕进行搜索,已经用的面积和体积 
{  
   int i,j,Maxv;
   if(floor==0&&v==0){if(s<Best)Best=s;return;}
   if(s+Mins[floor]>=Best)return;//剪枝1,最优性剪枝
   if(v<Minv[floor])return;//剪枝2,根据下界进行可行性剪枝
   Maxv=0;
   for(i=floor;i>=1;i--)
       Maxv+=(r-i)*(r-i)*(h-i);//最大方式做蛋糕
   if(Maxv<v)return; //剪枝3,根据上界进行可行性剪枝
   for(i=r-1;i>=floor;i--) //枚举floor层蛋糕的半径
      for(j=h-1;j>=floor;j--) //枚举floor层蛋糕的高度
         search(floor-1,i,j,s+2*i*j,v-i*i*j);
}
int main()
{  
   int r,h,s,v;
   Read();
   Best=0x7fffffff/2;
   for(r=m;r<=sqrt(n);r++)//枚举最底层蛋糕的半径
      for(h=n/(r*r);h>=m;h--)//枚举最底层蛋糕的对应高度
      {  
         s=r*r+2*r*h;
         v=n-r*r*h;
         search(m-1,r,h,s,v);
      }   
   cout<<Best<<endl;
}