题目链接:https://leetcode.cn/problems/ones-and-zeroes
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的长度,该子集中 最多 有 m 个 0 和 n 个 1 。
如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。:
示例1:
输入:strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3
输出:4
解释:最多有 5 个 0 和 3 个 1 的最大子集是 {"10","0001","1","0"} ,因此答案是 4 。
其他满足题意但较小的子集包括 {"0001","1"} 和 {"10","1","0"} 。{"111001"} 不满足题意,因为它含 4 个 1 ,大于 n 的值 3 。
如果大家对01背包问题有了解,本题乍一看运用简单的01背包问题确实无法解决,但如果大家能想到运用到两个维度的01背包问题,这题就可以解决了。
分析题意,首先要确定本题还是属于dp动态规划问题,那么就要确定dp数组的含义,dp[i][j],表示最多有i个0和j个1的最大子集的大小为dp[i][j],那么我们需要在每一层,也就是每遍历得到一个str的子集字符串,都要在在这一层统计0和1的个数并且更新dp数组,乍一听要三层for循环,有点害怕哈哈哈,在确定遍历顺序,也就是m和n,先遍历谁都无所谓,都是物品的一个维度
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
int dp[m+1][n+1];
//初始化dp数组
for(int i=0;i<=m;i++)
for(int j=0;j<=n;j++)
dp[i][j]=0; // 最多有i个0和j个1的strs的最大子集的大小为dp[i][j]
for(int i=0;i<strs.size();i++)
{
int oneNum=0; // 这一层0的个数
int zeroNum=0; // 这一层1的个数
string j=strs[i];
// 统计每一层1和0的个数
for(int k=0;k<j.length();k++)
{
if(j[k]=='0'){
zeroNum++;
}else{
oneNum++;
}
}
//更新dp数组
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);
}
}
}
return dp[m][n];
}
};