LeetCode 474 一和零（二维费用背包应用）

题目描述
给你一个二进制字符串数组 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;
}