二维费用背包问题+空间优化(滚动数组)

题目描述:

在计算机世界,我们一直追求用最小的资源产生最大的价值。

现在,假设你可以支配m个0和n个1。同时有一些只有0和1组成的字符串。

你的任务是用这些0和1去组成这些字符串,输出最多能组成多少个字符串。每个0和1只能被使用一次。

样例输入


样例一

输入: Array = {"10", "0001", "111001", "1", "0"}, m = 5, n = 3

输出: 4

解释: 用5个0和3个1可以至多组成4个给定字符串,分别为“10”、”0001”、”1”、”0”。

该题意可以转换为==》你有容量为m个0,n个1的背包,然后给你如何把字符串数组中的字符串放到该背包中,使得放的字符串数目尽可能地多?

然后就可以得出去状态转移方程:

dp[i][j][k]代表有j个0和k个1的背包,然后把前i个串放进去该背包中,所放的字符串数目的最大值

dp[i][j][k] = max{ dp[i-1][j][k](不放),dp[i-1][j-v[i]][k-u[i]]() }

代码如下:

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
#define maxn 128
string s[] = { "10", "0001", "111001", "1", "0" };
int m = 5;//0的数目
int n = 3;//1的数目
int zeroNum[maxn], oneNum[maxn];
int dp[maxn][maxn][maxn];
int main()
{
	for (int i = 1; i <= 5; i++)
	{
		for (int j = 0; j < s[i].length(); j++)
		{
			if (s[i][j] == '1'){
				oneNum[i]++;
			}
			else{
				zeroNum[i]++;
			}
		}
	}

	for (int i = 1; i <= 5; i++)
	{
		for (int j = 0; j <= m; j++)
		{
			for (int k = 0; k <= n; k++)
			{
				if (j >= zeroNum[i] && k >= oneNum[i]){
					dp[i][j][k] = max(dp[i - 1][j][k], dp[i-1][j - zeroNum[i]][k - oneNum[i]] + 1);
				}
				else{
					dp[i][j][k] = dp[i - 1][j][k];
				}
			}
		}
	}
	cout << dp[5][m][n] << endl;

	return 0;
}

空间优化(滚动数组):


由于前i个串的情况都是由前i-1个串推导出来的,与前1~i-2的情况没有任何的关系,这也就是无后效性(个人理解,不对求纠正),然后我们可以把这一维去掉,即状态转移方程变成:

dp[i][j][k] = max{ dp[i-1][j][k](不放),dp[i-1][j-v[i]][k-u[i]]() } =》dp[j][k] = max{ dp[j][k](不放),dp[j-v[i]][k-u[i]]() }(j、k的遍历顺序要变成倒序)

变成倒序的理由:因为dp[i][j][k]是由dp[i-1][j - zeroNum[i]][k - oneNum[i]]推导出来的,而不是dp[i][j - zeroNum[i]][k - oneNum[i]]
所以dp[i][j - zeroNum[i]][k - oneNum[i]]必须后于dp[i][j][k]求值才能保证dp[i][j][k]是由dp[i-1][j - zeroNum[i]][k - oneNum[i]]推导而来的


代码如下:

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
#define maxn 128
string s[] = { "10", "0001", "111001", "1", "0" };
int m = 5;//0的数目
int n = 3;//1的数目
int zeroNum[maxn], oneNum[maxn];
int dp[maxn][maxn];
int main()
{
	for (int i = 1; i <= 5; i++)
	{
		for (int j = 0; j < s[i].length(); j++)
		{
			if (s[i][j] == '1'){
				oneNum[i]++;
			}
			else{
				zeroNum[i]++;
			}
		}
	}

	for (int i = 1; i <= 5; i++)
	{
		for (int j = m; j >= zeroNum[i]; j--)
		{
			for (int k = n; k >= oneNum[i]; k--)
			{
				if (j >= zeroNum[i] && k >= oneNum[i]){
					dp[j][k] = max(dp[j][k], dp[j - zeroNum[i]][k - oneNum[i]] + 1);
				}
			}
		}
	}
	cout << dp[m][n] << endl;

	return 0;
}


  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值