dfs剪枝 Subtask#1 的第一个没过,其余全过,求调! 代码:
#include <bits/stdc++.h>
using namespace std;
int n,m,ans=0x3f3f3f3f;
int ss[20],vv[20];
void dfs(int k,int r,int h,int s,int v){
if(v+vv[k]>n) return;
int t=0;
for(int i=r-1,j=h-1;i>=1&&j>=1;i--,j--){
t+=i*i*j;
}
if(v+t<n) return;
if(s+ss[k]>=ans) return;
if(s+2*(n-v)/r>=ans) return;
if(k==m+1){
if(v==n) ans=min(ans,s);
return;
}
for(int i=m+1-k;i<r&&i*i<=n-v;i++){
for(int j=m+1-k;j<h&&v+i*i*j<=n;j++){
dfs(k+1,i,j,s+i*2*j,v+i*i*j);
}
}
}
int main(){
cin>>n>>m;
for(int i=1;i<=m;i++){
ss[i]=ss[i-1]+2*i*i;
vv[i]=vv[i-1]+i*i*i;
}
reverse(ss+1,ss+m+1);
reverse(vv+1,vv+m+1);
for(int r=m;r*r<=n;r++){
for(int h=m;r*r*h<=n;h++){
dfs(2,r,h,r*r+2*r*h,r*r*h);
}
}
cout<<ans;
return 0;
}