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;
}