leetcode 每日一题 474. 一和零
我的思路
我想的是用一个 map 存每个字符串中的 0,1。然后再用背包解决问题。最后还是没做出来,看了题解。
题解思路
官方题解使用了三维动态规划,三个维度分别是字符串、0 的容量和 1 的容量。定义三维 dp[i][j][k]表示在前 i 个字符串中,使用 j 个 0 和 k 个 1 能够得到的最多的字符串数量。最终答案为 dp[strs.length][m][n].
var findMaxForm = function (strs, m, n) {
const length = strs.length;
const dp = new Array(length + 1)
.fill(0)
.map(() => new Array(m + 1).fill(0).map(() => new Array(n + 1).fill(0)));
for (let i = 1; i <= length; i++) {
const zerosOnes = getZerosOnes(strs[i - 1]);
let zeros = zerosOnes[0],
ones = zerosOnes[1];
for (let j = 0; j <= m; j++) {
for (let k = 0; k <= n; k++) {
dp[i][j][k] = dp[i - 1][j][k];
if (j >= zeros && k >= ones) {
dp[i][j][k] = Math.max(
dp[i][j][k],
dp[i - 1][j - zeros][k - ones] + 1
);
}
}
}
}
return dp[length][m][n];
};
const getZerosOnes = (str) => {
const zerosOnes = new Array(2).fill(0);
const length = str.length;
for (let i = 0; i < length; i++) {
zerosOnes[str[i].charCodeAt() - "0".charCodeAt()]++;
}
return zerosOnes;
};
总结
三维动态规划,要仔细考虑清楚状态方程。动态规划还是需要系统学习一下。