UVALive3637(01背包+贪心)

链接：点击打开链接

题意：给出n本书的高和宽，要求做一个三层的书架，将所有书放入书架中，每层不能为空，每层的高度为当层最高的数的高度，当层的宽度为当层所有书的宽度和，书架的宽度为三层的最大宽度，书架的高度为三层的高度和，求书架高度乘宽度的最小值

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
    int h,w;
    friend bool operator<(node a,node b){
        return a.h>b.h;
    }
}s[105];
int dp[2505][2505];
int main(){                                     //dp[i][j]表示第二层和第三层宽为i和j
    int t,n,i,j,k,ans,sum;                      //的两层高的和最小是多少
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        sum=0;
        for(i=0;i<n;i++){
        scanf("%d%d",&s[i].h,&s[i].w);
        sum+=s[i].w;
        }
        sort(s,s+n);                            //按照高进行排序，因为最高的那一层
        memset(dp,INF,sizeof(dp));              //高已知，所以进行背包时可以不用算
        dp[0][0]=0;                             //最高的那个，那么推出两层后可以直
        for(i=1;i<n;i++){                       //接算出三层
            for(j=sum;j>=0;j--){
                for(k=sum;k>=0;k--){
                    if(j==0)                    //只有在一层刚放第一个时会增加高度
                    dp[j+s[i].w][k]=min(dp[j+s[i].w][k],dp[j][k]+s[i].h);
                    else
                    dp[j+s[i].w][k]=min(dp[j+s[i].w][k],dp[j][k]);
                    if(k==0)
                    dp[j][k+s[i].w]=min(dp[j][k+s[i].w],dp[j][k]+s[i].h);
                    else
                    dp[j][k+s[i].w]=min(dp[j][k+s[i].w],dp[j][k]);
                }
            }
        }
        ans=INF;
        for(i=1;i<=sum;i++)
        for(j=1;j<=sum;j++)
        if(dp[i][j]!=INF)                       //求出第一层的宽
        ans=min(ans,max(sum-i-j,max(i,j))*(dp[i][j]+s[0].h));
        printf("%d\n",ans);
    }
    return 0;
}