LeetCode第 474 题：一和零(C++)

最新推荐文章于 2021-12-06 16:21:15 发布

zj134_

最新推荐文章于 2021-12-06 16:21:15 发布

阅读量498

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_32523711/article/details/108174110

版权

leetcode 专栏收录该内容

334 篇文章 21 订阅

订阅专栏

474. 一和零 - 力扣（LeetCode）
在这里插入图片描述

二维费用的01背包问题，0和1就是每次选取消耗的两个费用，每个物品都会分别占用0和1的容量，而所有物品的价值可以认为都是1。

此题的状态应该是三维的：两维表示0和1的容量，1维表示可选择的物品，三维的dp如下：

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        //dp[i][j]表示i个0、j个1能够拼出的最大字符串个数
        vector<vector<vector<int>>> dp(strs.size() + 1, vector<vector<int>>(m + 1 , vector<int>(n + 1 , 0)));
        for(int k = 1; k <= strs.size(); ++k){//对于每个字符串
            int zero = 0, one = 0;
            for(const auto &c : strs[k-1])  c == '0' ? ++zero : ++one;
            for(int i = 0; i <= m; ++i){
                for(int j = 0; j <= n; ++j){
                    if(i - zero >= 0 && j - one >= 0)
                        dp[k][i][j] = max(dp[k-1][i-zero][j-one] + 1, dp[k-1][i][j]);//选或不选
                    else
                        dp[k][i][j] = dp[k-1][i][j];//继承之前的值
                }
            }
        }
        return dp[strs.size()][m][n];
    }
};

01背包常规状态压缩，注意逆序处理：

class Solution {
public:
    int dp[110][110];
    int findMaxForm(vector<string>& strs, int m, int n) {
        for(const auto &s : strs){
            int zero = 0, one = 0;
            for(const auto &c : s)  c == '0' ? ++zero : ++one;
            for(int i = m; i >= zero; --i){
                for(int j = n; j >= one; --j)   dp[i][j] = max(dp[i][j], 1+dp[i-zero][j-one]);
            }
        }
        return dp[m][n];
    }
};