1105. 填充书架
————————————————————————————————————————————
附近的家居城促销,你买回了一直心仪的可调节书架,打算把自己的书都整理到新的书架上。
你把要摆放的书 books 都整理好,叠成一摞:从上往下,第 i 本书的厚度为 books[i][0],高度为 books[i][1]。
按顺序 将这些书摆放到总宽度为 shelf_width 的书架上。
先选几本书放在书架上(它们的厚度之和小于等于书架的宽度 shelf_width),然后再建一层书架。重复这个过程,直到把所有的书都放在书架上。
需要注意的是,在上述过程的每个步骤中,摆放书的顺序与你整理好的顺序相同。 例如,如果这里有 5 本书,那么可能的一种摆放情况是:第一和第二本书放在第一层书架上,第三本书放在第二层书架上,第四和第五本书放在最后一层书架上。
每一层所摆放的书的最大高度就是这一层书架的层高,书架整体的高度为各层高之和。
以这种方式布置书架,返回书架整体可能的最小高度。
输入:books = [[1,1],[2,3],[2,3],[1,1],[1,1],[1,1],[1,2]], shelf_width = 4
输出:6
解释:
3 层书架的高度和为 1 + 3 + 2 = 6 。
第 2 本书不必放在第一层书架上。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/filling-bookcase-shelves
————————————————————————————————————————————
解题思路:使用动态规划,假设dp[i]代表当前书架中i本书的最小高度。对于第i+1本书,最坏的情况是在dp[i]的高度下加上第i+1本书的高度,这样可以进行比较。对于第i本书,考虑第i本书和第i+1本书在同一层的书架的高度,这时候如果宽度仍然符合要求,则可以考虑将i+1和i和i-1本书放在一起书架的高度,依次类推,直到书架宽度超出要求,则可以得到i+1本书的最低高度。其C++代码如下:
class Solution {
public:
int minHeightShelves(vector<vector<int>>& books, int shelf_width) {
int length = books.size();
vector<int> dp(length+1,0);
dp[1]=books[0][1]; #保存第一本书的书架的高度
for(int i=1;i<length;i++)
{
dp[i+1] = books[i][1] + dp[i]; #第i本书时的最坏高度的情况
int wide = books[i][0]; #设定动态规划时的初始宽度
int high = books[i][1]; #设定动态规划时的初始高度
for(int j=i;j>0;j--)
{
wide = wide + books[j-1][0];
if(wide>shelf_width) #如果此时书架的宽度仍然符合要求,则进行动态规划
break;
high = max(high,books[j-1][1]); #选择这一层中的最高高度
dp[i+1] = min(dp[i+1],dp[j-1] + high); #进行判断,选择书架高度最低的数值
}
}
return dp[length];
}
};