poj 1190 生日蛋糕 剪枝dfs

两个可行性剪枝 if(i*i*j*m

                int tmp=0,p=1;
                for(int k=1;k<=m;k++)
                {
                  tmp+=p*p*p;
                    p++;
                }
               if(tmp>n) continue;
一个最优化剪枝 if(2*n/i+s>=ans) continue;

 

 

#include <iostream>

#include<stdio.h>

#include<string.h>

#include<math.h>

using namespace std;

const int inf=999999;

int ans=inf;

int M;

int dfs(int r,int h,int n,int m,int s)

{

   for(int i=r-1;i>=1;i--)

   for(int j=h-1;j>=1;j--)

   if(i*i*j<=n)

   {

       if(m==1)

       {

           int tmp=0;

           if(m==M) tmp=i*i;

           if(i*i*j!=n) continue;

           if(s+2*i*j+tmp<ans) ans=s+2*i*j+tmp;

           continue;

       }

       if(i*i*j*m<n) continue;

       {

           int tmp=0,p=1;

           for(int k=1;k<=m;k++)

           {

               tmp+=p*p*p;

               p++;

           }

           if(tmp>n) continue;

       }

       if(m!=M)

       {

           if(2*n/i+s>=ans) continue;

       }

       int tmp=0;

       if(m==M) tmp=i*i;

       dfs(i,j,n-i*i*j,m-1,s+tmp+2*i*j);

   }

   return(0);

}

int main()

{

    int n,m;

    scanf("%d %d",&n,&m);

    M=m;

    int r=sqrt(n/m)+1;

    int h=n/m/m+1;

    dfs(r,h,n,m,0);

    if(ans!=inf)

    printf("%d\n",ans);

    else

    printf("0\n");

    return 0;

}