题目描述
给你一个二进制字符串数组 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 。
题解:二维费用背包,将0的个数看作是重量,将1的个数看做是体积,套用二维费用背包的模板,首先需要计算每个字符串中所包含的0和1的个数,也就是换成二维费用背包问题中的重量和体积。使用了空间优化方式。
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
vector<vector<int>> f(m+1,vector<int>(n+1,0));
for(int i=0;i<strs.size();i++)
{
int a=0,b=0;
for(auto x:strs[i])
{
if(x=='0') a++;
else b++;
}
for(int j=m;j>=a;j--)
{
for(int k=n;k>=b;k--)
{
f[j][k]=max(f[j][k],f[j-a][k-b]+1);
}
}
}
return f[m][n];
}
};
一维背包优化需要体积从大到小循环。
#include<iostream>
using namespace std;
const int N=1010;
int n,m;
int v[N],w[N];
int f[N];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
for(int i=1;i<=n;i++)
{
for(int j=m;j>=v[i];j--)
{
if(j>=v[i])
{
f[j]=max(f[j],f[j-v[i]]+w[i]);
}
}
}
cout<<f[m]<<endl;
return 0;
}
二维背包优化,体积重量均从大到小循环。
#include<bits/stdc++.h>
using namespace std;
const int N=110;
int n,v,m;
int f[N][N];
int main()
{
cin>>n>>v>>m;
for(int i=0;i<n;i++)
{
int a,b,c;
cin>>a>>b>>c;
for(int j=v;j>=a;j--)
{
for(int k=m;k>=b;k--)
{
f[j][k]=max(f[j][k],f[j-a][k-b]+c);
}
}
}
cout<<f[v][m]<<endl;
}