ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode 474 Ones and Zeroes 滚动数组优化DP

2022-06-04 20:34:44  阅读:162  来源: 互联网

标签:Zeroes subset set int 个数 strs Ones 474 dp


You are given an array of binary strings strs and two integers m and n.

Return the size of the largest subset of strs such that there are at most m \(0\)'s and n \(1\)'s in the subset.

A set \(x\) is a subset of a set \(y\) if all elements of \(x\) are also elements of \(y\).

Solution

考虑朴素的转移,设 \(dp[i][j][k]\) 表示在前 \(i\) 个字符串中,\(0\) 个数上限为 \(j\),\(1\) 个数上限为 \(k\) 时的最大子集个数。那么如何转移呢?类似于背包问题:我们或者不选当前的,也就是从上一层继承答案;或者选择当前的,进行答案的更新:

\[dp[i][j][k] = \max(dp[i-1][j][k],dp[i][j-n_0][k-n_1]+1) \]

其中 \(n_0,n_1\) 分别代表当前串中 \(0,1\) 的个数。

观察一下,我们此时的答案,只需要从上一层转移。所以用滚动数组来滚去一维。但此时需要注意的是,我们需要倒序来更新,否则就会将之前的上一层给抹去。

点击查看代码
class Solution {
private:
    int dp[102][102];
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int l = strs.size();
        for(int i=0;i<l;i++){
            int n0 = 0, n1 = 0;
            for(int j=0;j<strs[i].length();j++){
                if(strs[i][j]=='0')n0++;
                else n1++;
            }
            for(int j=m;j>=n0;j--){
                for(int k=n;k>=n1;k--){
                    dp[j][k] = max(dp[j][k], dp[j-n0][k-n1]+1);
                }
            }
        }
        return dp[m][n];
    }
};

标签:Zeroes,subset,set,int,个数,strs,Ones,474,dp
来源: https://www.cnblogs.com/xinyu04/p/16342643.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有