POJ-3124-The Bookcase

这个题大意是说有n本书,他们有各自的高度和宽度,现在有一个3层的书架,然后让你把这些书放到书架上,然后求出h(3层书架中每层最高的书高度之和)与w(每层宽度之和的最大值),使得 w*h最小。

思路:这个题明显是dp,首先对所有书按高度进行排序,并且默认第一层的最高的书大于第二层最高的,同理第二层大于第三层,用dp[i][j[k]表示当前放了i本书,第二层的宽度为j,第三层的宽度为k的最小高度。那么对于不同情况进行枚举状态就行了。不过需要注意的是这里如果直接这样开数组明显太大了,需要用滚动数组,还有一点就是需要不断的优化,不然肯定TLE

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=1<<29;
const int maxn=71;
const int maxm=31*maxn;
struct Book
{
    int width;
    int height;
    bool operator < (const Book &a)const
    {
        if(height==a.height)
            return width<a.width;
        return height>a.height;
    }
}a[maxn];
int n,sum[maxn],dp[2][maxm][maxm];
void Init(int pos)
{
    for(int j=0;j<=sum[n];j++)
        for(int k=0;j+k<=sum[n];k++)
            dp[pos][j][k]=inf;
}
void solve()
{
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i].width;
    Init(1);
    dp[1][0][0]=a[1].height;
    int pos=0;
    for(int i=2;i<=n;i++,pos^=1)
    {
        Init(pos);
        for(int j=0;j<=sum[i];j++)
            for(int k=0;k<=j+30&&j<=sum[n]-j-k+30&&j+k<=sum[i];k++)
                if(dp[pos^1][j][k]!=inf)
                {
                    dp[pos][j][k]=min(dp[pos][j][k],dp[pos^1][j][k]);
                    if(!j)
                        dp[pos][a[i].width][k]=min(dp[pos][a[i].width][k],dp[pos^1][j][k]+a[i].height);
                    if(j)
                        dp[pos][j+a[i].width][k]=min(dp[pos][j+a[i].width][k],dp[pos^1][j][k]);
                    if(j&&!k)
                        dp[pos][j][a[i].width]=min(dp[pos][j][a[i].width],dp[pos^1][j][k]+a[i].height);
                    if(k)
                        dp[pos][j][k+a[i].width]=min(dp[pos][j][k+a[i].width],dp[pos^1][j][k]);
                }
    }
    long long ans=1LL<<60;
    for(int i=1;i<=sum[n];i++)
        for(int j=1;i+j<=sum[n];j++)
            if(dp[pos^1][i][j]!=inf)
            {
                int w=max(i,max(j,sum[n]-i-j));
                ans=min(ans,(long long)w*dp[pos^1][i][j]);
            }
    printf("%lld\n",ans);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d%d",&a[i].height,&a[i].width);
        sort(a+1,a+n+1);
        solve();
    }
    return 0;
}


你可能感兴趣的:(dp)