面试题 08.13. 堆箱子
堆箱子。给你一堆n个箱子,箱子宽 wi、深 di、高 hi。箱子不能翻转,将箱子堆起来时,下面箱子的宽度、高度和深度必须大于上面的箱子。实现一种方法,搭出最高的一堆箱子。箱堆的高度为每个箱子高度的总和。
输入使用数组[wi, di, hi]表示每个箱子。
示例1:
输入:box = [[1, 1, 1], [2, 2, 2], [3, 3, 3]]
输出:6
示例2:
输入:box = [[1, 1, 1], [2, 3, 4], [2, 6, 7], [3, 4, 5]]
输出:10
提示:
箱子的数目不大于3000个。
题解
排序后按最长上升子序列套模板。
AC代码
class Solution {
public:
struct Node
{
int w,d,h;
}a[3010];
static int cmp(Node a1,Node a2)
{
if(a1.w<a2.w)return true;
else if(a1.w==a2.w)
{
if(a1.d<a2.d)return true;
if(a1.d==a2.d)return a1.h<a2.h;
}
return false;
}
int dp[3010];
int pileBox(vector<vector<int>>& box)
{
for(int i=0;i<box.size();i++)
{
a[i].w=box[i][0];
a[i].d=box[i][1];
a[i].h=box[i][2];
}
sort(a,a+box.size(),cmp);
int res=0;
for(int i=0;i<box.size();i++)
{
dp[i]=a[i].h;
for(int j=i-1;j>=0;j--)
{
if(a[i].w>a[j].w&&a[i].d>a[j].d&&a[i].h>a[j].h)
{
dp[i]=max(dp[i],dp[j]+a[i].h);
}
}
res=max(res,dp[i]);
}
return res;
}
};