poj 1190(dfs剪枝)

最新推荐文章于 2018-10-04 23:30:34 发布

A_Happyer

最新推荐文章于 2018-10-04 23:30:34 发布

阅读量564

点赞数

分类专栏：搜索

本文链接：https://blog.csdn.net/u014569598/article/details/41623273

版权

搜索专栏收录该内容

44 篇文章 0 订阅

订阅专栏

先初始化算出n层蛋糕所需的最小体积，用其剪枝，可以大大提升速度。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<map>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rev(i,a,b) for(int i=(a);i>=(b);i--)
#define clr(a,x) memset(a,x,sizeof a)
#define inf 0x3f3f3f3f
typedef long long LL;
using namespace std;

const int maxn=25;
int n,m,rr,hr,ans,mmin[25];

void init()
{
    mmin[0]=0;int cc=1;
    for(int i=1;i<23;i++)
    {
        mmin[i]=mmin[i-1]+i*i*(cc++);
    }
}
void dfs(int a,int b,int sum,int num,int place)
{
    if(sum+a*a*b*(m-place)<n)return ;
    if(num>ans)return ;
    if(sum+mmin[m-place]>n)return ;
    if(place==m)
    {
        if(sum==n)ans=min(ans,num);
        return ;
    }
    for(int i=a-1;i>=m-place;i--)
        for(int j=b-1;j>=m-place;j--)
        {
            dfs(i,j,sum+j*i*i,num+2*i*j,place+1);
        }
}
int main()
{
    init();
    while(~scanf("%d%d",&n,&m))
    {
        ans=inf;
        for(int j=m;j<=n;j++)
            for(int i=m;i<=sqrt(n/j);i++)
                dfs(i,j,j*i*i,i*i+2*i*j,1);
        if(ans!=inf)printf("%d\n",ans);
        else printf("0\n");
    }
    return 0;
}