Leetcode474. 一和零-代码随想录

题目：

---

代码(首刷看解析 2024年2月26日）

class Solution {
public:
    // 二维 0 1背包
    int findMaxForm(vector<string>& strs, int m, int n) {
        // 1 二维 [i]表示 0 的个数，上限m; [j]表示 1 的个数，上限n
        vector<vector<int>> dp(m + 1, vector<int>(n + 1, 0));
        // 2 初始化 TODO:
        dp[0][0] = 0;
        // 3 遍历 TODO:倒序遍历[i][j]
        for (auto str : strs) {
            int zeroNum = 0;
            int oneNum = 0;
            for (int it : str) {
                if (it == '0') zeroNum++;
                else if(it == '1')oneNum++;
            }
            cout<<"zero:"<<zeroNum<<"  one:"<<oneNum<<endl;
            for (int i = m; i >= zeroNum; --i) {
                for (int j = n; j >= oneNum; --j) {
                    dp[i][j] = max(dp[i][j], 
                                dp[i - zeroNum][j - oneNum] + 1);
                    cout<<"dp["<<i<<"]"<<"["<<j<<"]:"<<dp[i][j]<<"  ";
                }
                cout<<endl;
            }
            cout<<"___________________________________"<<endl;
        }
        return dp[m][n];
            // 4 递推公式 dp[i][j] = max(dp[i - zeroNum][j - oneNum], 
            // dp[i - zeroNum][j - oneNum] + 1);

    }
};

---

代码(二刷看了眼思路 2024年7月9日 go)

// strs[i]有x个0 y个1,这两个都是重量
func findMaxForm(strs []string, m int, n int) int {
    // 1. dp[m][n]表示m个0和n个1的最大子集长度
    // 2. 递推公式:dp[m][n] = max(dp[m - x][n - y] + 1, dp[m][n])
    // 3. dp[0][0] = 0
    dp := make([][]int, m + 1)
    for i := range dp {
        dp[i] = make([]int, n + 1)
    }
    dp[0][0] = 0
    // 获得每个str的0和1的个数
    nums := make([][]int, len(strs))
    for i := range nums {
        nums[i] = make([]int, 2)
    }
    for i := 0; i < len(nums); i++ {
        nums[i][0] = getZeroNum(strs[i])     //strs[i] '0'的数量
        nums[i][1] = len(strs[i]) - nums[i][0] //strs[i] '1'的数量
    }
    // 4. 遍历顺序:物品从左往右,0从右往左,1从右往左
    for i := 0; i < len(strs); i++ {
        for j := m; j >= nums[i][0]; j-- {
            for k := n; k >= nums[i][1]; k-- {
                dp[j][k] = max(dp[j - nums[i][0]][k - nums[i][1]] + 1, dp[j][k])
            }
        }
    }
    return dp[m][n]
}

func getZeroNum(str string) int {
    count := 0
    for _, v := range str {
        if v == '0' {
            count++
        }
    }
    return count
}

func max(a,b int) int {
    if a > b {
        return a
    }
    return b
}