算法之使用动态规划解决填充书架问题
给定一个数组 books ,其中 books[i] = [thicknessi, heighti] 表示第 i 本书的厚度和高度。你也会得到一个整数 shelfWidth 。
按顺序 将这些书摆放到总宽度为 shelfWidth 的书架上。
先选几本书放在书架上(它们的厚度之和小于等于书架的宽度 shelfWidth ),然后再建一层书架。重复这个过程,直到把所有的书都放在书架上。
需要注意的是,在上述过程的每个步骤中,摆放书的顺序与给定图书数组 books 顺序相同。
例如,如果这里有 5 本书,那么可能的一种摆放情况是:第一和第二本书放在第一层书架上,第三本书放在第二层书架上,第四和第五本书放在最后一层书架上。
每一层所摆放的书的最大高度就是这一层书架的层高,书架整体的高度为各层高之和。
以这种方式布置书架,返回书架整体可能的最小高度。
示例 1:
输入:books = [[1,1],[2,3],[2,3],[1,1],[1,1],[1,1],[1,2]], shelfWidth = 4
输出:6
解释:
3 层书架的高度和为 1 + 3 + 2 = 6 。
第 2 本书不必放在第一层书架上。
示例 2:
输入: books = [[1,3],[2,4],[3,2]], shelfWidth = 6
输出: 4
提示:
1 <= books.length <= 1000
1 <= thicknessi <= shelfWidth <= 1000
1 <= heighti <= 1000
解题思路
假设i是books列表的最后一本书,并放在最后一层书架上。
let curWidth = 0, curWidth 表示当前这一层的宽度,从i往前依次遍历 i--,
let curWidth += book[i[[0]。
那么这最后一层书架能放的书是只要满足 curWidth <= shelfWidth ,就能和第i本书放在同一层。
let maxHeight = 0,maxHeight 表示当前这一层的最大高度,
那么最后一层的最大高度等于Mth.max(maxHeight ,books[i][1])。
用dp[i]表示前i本书的最小高度之和,如果用一个变量j来表示第i本书所在层的第一本书,
此时j满足 0<= j <= i,那么前i本书的高度之和 = dp[j] + maxHeight 。
那么就是需要求 前j本书的最小高度之和 dp[j] ,由于j是不断变化的,
通过Math.min(dp[i],dp[j]+maxheight)来确定前i本书的最小高度之和。
代码部分
var minHeightShelves = function (books, shelfWidth) {
let len = books.length
// 因为第一本书要从下标1开始取,所以要创建len+1个元素
let dp = new Array(len + 1).fill(Infinity) // [Infinity,Infinity,Infinity,Infinity,Infinity,Infinity,Infinity,Infinity]
// 0本书的最小高度是0
dp[0] = 0 // [0,Infinity,Infinity,Infinity,Infinity,Infinity,Infinity,Infinity]
// dp[i] 表示第i本书之前的最小高度之和
// i <= len 要计算到列表最后一本书的最小高度之和
for (let i = 1; i <= len; i++) {
let curWidth = 0,
maxheight = 0
// 从第i本书开始,往前遍历,计算出第j到第i本书的最小高度之和,且满足第j到第i本书的宽度 = shelfWidth
for (let j = i - 1; j >= 0; j--) {
const [w, h] = books[j]
curWidth += w
if (curWidth > shelfWidth) {
break
}
// 当前层的最高值
maxheight = Math.max(maxheight, h)
dp[i] = Math.min(dp[i], dp[j] + maxheight)
}
}
return dp[len]
}