leetcode 474. 一和零

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;
};

总结

三维动态规划，要仔细考虑清楚状态方程。动态规划还是需要系统学习一下。