又见神剪枝啊,不加
2*v/R+tmp>=ans
这句话的话,是无法避免超时的
它的意思是:当前解+剩余资源的最优解>=最优解时,剪枝,有点像a*的估价函数
#include <iostream>
#include <cmath>
using namespace std;
int R,H,n,m,ans,mv[21],tmp;
void dfs(int v,int dep,int R,int H) {
if (dep==0) {
if (v==0 && tmp<ans) ans=tmp;
return;
}
if (v-mv[dep-1]<0 || tmp>=ans || 2*v/R+tmp>=ans) return;
for (int r=R-1;r>=dep;r--) {
int Hm = min(H-1,(v-mv[dep-1])/r/r);
for (int h=Hm;h>=dep;h--)
if ((v-r*r*h)>=0) {
if (dep==m) tmp=r*r;
tmp+=2*r*h;
dfs(v-r*r*h,dep-1,r,h);
tmp-=2*r*h;
if (dep==m) tmp=0;
}
}
}
int main() {
memset(mv,0,sizeof(mv));
for (int i=1;i<=m;i++)
mv[i]=mv[i-1]+i*i*i;
while (~scanf("%d%d",&n,&m)) {
ans=10000000;
dfs(n,m,n+1,n+1);
if (ans==10000000) printf("0\n");
else printf("%d\n",ans);
}
return 0;
}