大体题意:给定一组书籍的厚度和高度,要全部放到书架上,使得书架总体的高度最小。默认书籍都是竖着放的,不会躺着放!
具体题目可以去力扣官网看一下
有个地方需要特别注意:题目说了物体是按照顺序摆放的,也就是说我在第一层可以放物体1+物体2+...,但是不能放物体1+物体3...。降低了题的难度,否则这道题估计就是hard了
这道题明显是一道动态规划,然而还是不会做,找了一道排名靠前且代码清晰的解,学习一下
const int INF = 1e9 + 5;
class Solution {
public:
int minHeightShelves(vector<vector<int>>& books, int shelf_width) {
int n = books.size();
vector<int> dp(n + 1, INF);
dp[0] = 0;
for (int i = 0; i < n; i++) {
int width = 0, height = 0;
for (int j = i; j < n; j++) {
width += books[j][0];
height = max(height, books[j][1]);
if (width <= shelf_width)
dp[j + 1] = min(dp[j + 1], dp[i] + height);
else break;
}
}
return dp[n];
}
};
看过以后就稍微明白了,原本的状态应该是dp[i][j],意思是在物体i上面放置物体j所需要的高度,而这个解答对空间做了优化所以就变成了一维dp。
第二层for循环的意思是:我们依次将书籍纳入这一层,并更新状态,直到这一层的宽度达到上限。
状态转移方程相对好理解的多:dp[j + 1] = min(dp[j + 1], dp[i] + height); 即物体j的高度,取之前的高度,以及在i上放置j转移来的高度的最小者。
下面是我根据上面的解答写出来的未经过空间优化的,可能会好理解一些?
const int INF = 1e9 + 5;
class Solution {
public:
int minHeightShelves(vector<vector<int>>& books, int shelf_width) {
int n = books.size();
vector<vector<int>> dp(n, vector<int>(n + 1,INF));
for(int i=0;i<=n;i++)
{
int height=0;
int width=0;
for(int j=i;j<n;j++)
{
width+=books[j][0];
height=max(height,books[j][1]);
if(width<=shelf_width&&i>0)
dp[i][j+1]=min(dp[i-1][j+1],dp[i-1][i]+height);
else if(width<=shelf_width&&i==0)
dp[i][j+1]=height;
else if(width>shelf_width)
break;
}
}
return dp[n-1][n];
}
};
这道题大体来说就是这样了,虽然之前总结过类似的背包问题,但是看来火候还差点远啊!