poj1190生日蛋糕--DFS

#include<stdio.h>
int r[10001];
int h[10001];
int N,M,S;
int miv[21];
int mis[21];
void dfs(int smm,int v,int dep,int h,int r){
    if(v>N) return;
    if(dep==0){
        if(v==N&&smm<S){
            S=smm;
            return;
        }
    }
    if(v+miv[dep-1]>N||smm+mis[dep-1]>=S||(N-v)/r*2+smm>S) return;
    for(int i=r-1;i>=dep;i--){
        if(dep==M) smm=i*i;

        int mxh;
        if((N-v-miv[dep-1])/(i*i)>h-1)
            mxh=h-1;
        else mxh=(N-v-miv[dep-1])/(i*i);
        for(int j=mxh;j>=dep;j--){
            dfs(smm+2*i*j,v+i*i*j,dep-1,j,i);
        }
    }
}
void mx(){
    for(int i=1;i<=20;i++){
        miv[i]=miv[i-1]+i*i*i;
        mis[i]=mis[i-1]+2*i*i;
    }
}
int main(){
    scanf("%d %d",&N,&M);
    S=99999999;
　　 mx();
    dfs(0,0,M,N+1,N+1);
    if(S==99999999) S=0;
    printf("%d\n",S);
    return 0;
}