题面

传送门

Solution

搜索每一层的半径和高度，然后加入一些剪枝就好了。

#include<stdio.h>

#include<stdlib.h>

#include<string.h>

#include<math.h>

#include<algorithm>

#include<queue>

#include<iostream>

#include<map>

#define ll long long

#define file(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout);

using namespace std;

inline int gi(){

	int sum=0,f=1;char ch=getchar();

	while(ch>'9' || ch<'0'){if(ch=='-')f=-1;ch=getchar();}

	while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}

	return f*sum;

}

const int Inf=2e9+10;

int n,m,ans=Inf,a[210],b[210];

void dfs(int v,int s,int p,int r,int h){

	if(p==0){

		if(v==n)ans=min(ans,s);

		return;

	}

	if(v+b[p-1]>n)return;

	if(s+a[p-1]>ans)return;

	if(2*(n-v)/r+s>=ans)return;

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

		if(p==m)s=i*i;

		int Maxh=min(h-1,(n-v-b[p-1])/(i*i));

		for(int j=Maxh;j>=p;--j)

			dfs(v+i*i*j,s+2*i*j,p-1,i,j);

	}

}

int main(){

	int i,j,k;

	n=gi();m=gi();

	a[0]=0;b[0]=0;

	for(i=1;i<=28;i++){

		a[i]=a[i-1]+2*i*i;

		b[i]=b[i-1]+i*i*i;

	}

	dfs(0,0,m,n+1,n+1);

	printf("%d\n",ans==Inf?0:ans);

	return 0;

}