码迷,mamicode.com
首页 > 其他好文 > 详细

动态规划小结——背包问题

时间:2015-02-12 16:23:47      阅读:408      评论:0      收藏:0      [点我收藏+]

标签:uva   动态规划   

背包问题是动态规划的经典问题,所有初次学习动态规划都会讲述这个经典问题。因此,有必要弄清跟背包问题的所有分析过程并熟练掌握各种类型的代码。

一,完全背包问题

1.问题描述:有n种物品,每种物品有无限多个,第i个物品重量是wi,价值是vi,从这些物品中挑选总重量不超过W的物品,求出挑选物品价值的最大值。

   限制条件:1≤n≤100,1≤wi,vi≤100,1≤W≤10000

2.解题思路:本题类似于“硬币问题”,硬币问题只要求凑够相应的面值,这里只不过多了一个新的属性——价值,即不仅要凑够重量,还要使得价值尽可能大。按照动态规划的分析思路,先定义状态d(j):剩余重量为j时的最大价值。那么如果当前使用了第i个物品(如果可以用的话),状态变转移到j-w[i],而价值变为d[j-w[i]]+v[i],因此状态转移方程如下:

d[j]=max(d[j],d[j-w[i]]+v[i]);(0≤i<n)

这个过程中随着物品的增多,剩余的重量在逐渐减小。

3.代码:

(1)不包含打印路径

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn];
int w[maxn],v[maxn];
int n, W;
int dp(int j)
{
	int&ans = d[j];
	if (ans>0)return ans;
	for (int i = 0; i < n;i++)
	if (j >= w[i])
		ans = max(ans, dp(j-w[i]) + v[i]);
	return ans;
}
int main()
{
	//freopen("test.txt", "r", stdin);
	scanf("%d%d", &n, &W);
	rep(i, n)
		scanf("%d", w + i);
	rep(i, n)
		scanf("%d", v + i);
	dp(W);//从状态W开始,走到状态0
	cout << d[W] << endl;
	return 0;
}

(2)包含打印路径

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn];
int w[maxn],v[maxn];
int n, W;
int dp(int j)
{
	int&ans = d[j];
	if (ans>0)return ans;
	for (int i = 0; i < n;i++)
	if (j >= w[i])
		ans = max(ans, dp(j-w[i]) + v[i]);
	return ans;
}
void print_ans(int j)
{
	for (int i = 0; i < n;i++)
	if (j >= w[i] && d[j] == d[j - w[i]] + v[i])
	{
		printf("%d ", i);
		print_ans(j - w[i]);
		break;
	}
}
int main()
{
	//freopen("test.txt", "r", stdin);
	scanf("%d%d", &n, &W);
	rep(i, n)
		scanf("%d", w + i);
	rep(i, n)
		scanf("%d", v + i);
	dp(W);//从状态W开始,走到状态0
	cout << d[W] << endl;
	print_ans(W);//打印所用物品的序号
	puts("");
	return 0;
}


二,0-1背包问题

1.问题描述:有n种物品,每种只有一个,,第i个物品重量是wi,价值是vi,从这些物品中挑选总重量不超过W的物品,求出挑选物品价值的最大值。

   限制条件:1≤n≤100,1≤wi,vi≤100,1≤W≤10000

2.解题思路:现在的问题又增加了一个限制条件——每种物品的个数,可以发现,原来的状态转移方程已经不适用了,因为只凭借“剩余重量的大小”无法得知每个物品是否用过,因此需要修改原来定义的状态,使得以后的决策可以有序化。

想象所有的物品按照序号摆成一排,现在,定义d(i,j)表示当前正在第i个物品处,背包的剩余容量为j时的最大价值。显然,这时的决策有两个:(1)不使用物品i;(2)使用物品i;因此,状态转移方程如下:

d(i,j)=max(d(i+1,j),d(i+1,j-w[i])+v[i]);(j≥w[i])

边界:d(n-1,j)=0;

上式中,i要逆序枚举,因为公式中i时候的状态是由i+1时候的得到的。上式中的j随着i的减小而增大,这也不难由公司看出。最后的答案是d[0][W],这样代码便不难写出。

3.代码:

(1)不包含打印解

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn][maxn];
int w[maxn],v[maxn];
int n, W;

int main()
{
	//freopen("test.txt", "r", stdin);
	memset(d, -1, sizeof(d));
	scanf("%d%d", &n, &W);
	rep(i, n)
		scanf("%d", w + i);
	rep(i, n)
		scanf("%d", v + i);
	for (int i = n - 1; i >= 0;i--)//i逆序枚举,由递推公式决定
	for (int j = 0; j <= W; j++)//j的顺序无所谓
	{
		d[i][j] = (i == n - 1 ? 0 : d[i + 1][j]);
		if (j >= w[i])
			d[i][j] = max(d[i][j], d[i + 1][j - w[i]] + v[i]);
	}
	printf("%d\n", d[0][W]);
	return 0;
}

(2)包含打印解

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn][maxn];
int w[maxn],v[maxn];
int vis[maxn];
int n, W;

int main()
{
	//freopen("test.txt", "r", stdin);
	memset(d, -1, sizeof(d));
	scanf("%d%d", &n, &W);
	rep(i, n)
		scanf("%d", w + i);
	rep(i, n)
		scanf("%d", v + i);
	for (int i = n - 1; i >= 0; i--)//i逆序枚举,由递推公式决定
	for (int j = 0; j <= W; j++)//j的顺序无所谓
	{
		d[i][j] = (i == n - 1 ? 0 : d[i + 1][j]);
		if (j >= w[i])
		d[i][j] = max(d[i][j], d[i + 1][j - w[i]] + v[i]);
	}
	printf("%d\n", d[0][W]);
	printf("Solutions :");//打印已经选择的物品
	int R = W;
	for (int i = 0; i < n;i++)
	for (int j = R; j >= 0;j--)
	if (d[i][j] == d[i + 1][j])break;
	else if (j >= w[i] && d[i][j] == d[i + 1][j - w[i]] + v[i])
	{
		printf(" %d", i);
		R -= w[i];
		break;
	}
	printf("\n");
	return 0;
}


三,0-1背包问题的规划方向

刚刚的状态d(i,j)的定义是: 当前正在第i个物品处,背包的剩余容量为j时的最大价值。这样的定义导致了枚举i时必须逆序。其实还有一种“对称”的状态定义:当前正在第i个物品处,背包的总容量为j时的最大价值。按照这样的定义,不难得到如下的状态转移方程:

d(i,j)=max(d(i-1,j),d(i-1,j-w[i])+v[i])(j≥w[i]);

边界:d(0,j)=0;

按照上式,如果选择了物品i,那么在i-1处背包的容量必须是j-w[i],这样才能装下i物品。由上式可知,i必须顺序枚举,因为i处的状态是由i-1处的状态得到的。而且j随着i的增大而增大,这和之前说的剩余容量在逐渐减小是一致的。最终的答案是d(n-1,W)代码与上面的类似。

3.代码:

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn][maxn];
int w[maxn],v[maxn];
int vis[maxn];
int n, W;

int main()
{
	//freopen("test.txt", "r", stdin);
	memset(d, -1, sizeof(d));
	scanf("%d%d", &n, &W);
	rep(i, n)
		scanf("%d", w + i);
	rep(i, n)
		scanf("%d", v + i);
	for (int i = 0; i < n; i++)//i顺序枚举,由递推公式决定
	for (int j = 0; j <= W; j++)//j的顺序无所谓
	{
		d[i][j] = (i == 0 ? 0 : d[i - 1][j]);
		if (j >= w[i])
		d[i][j] = max(d[i][j], d[i - 1][j - w[i]] + v[i]);
	}
	printf("%d\n", d[n - 1][W]);
	return 0;
}


其实,这种对称的定义方法还有一个额外的好处,就是允许边读入边计算,而不必把w,v保存下来。

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn][maxn];
int w, v;
int vis[maxn];
int n, W;

int main()
{
	//freopen("test.txt", "r", stdin);
	memset(d, -1, sizeof(d));
	scanf("%d%d", &n, &W);
	for (int i = 0; i < n; i++)//i顺序枚举,由递推公式决定
	{
		scanf("%d%d", &w, &v);//读入第i个物品的重量,价值
		for (int j = 0; j <= W; j++)
		{
			d[i][j] = (i == 0 ? 0 : d[i - 1][j]);
			if (j >= w)d[i][j] = max(d[i][j], d[i - 1][j - w] + v);
		}
	}
	printf("%d\n", d[n - 1][W]);
	return 0;
}


不仅如此,甚至可以把d数组降维,将其变成一维数组。利用原来的结果产生新的结果,节约内存空间。这是因为在更新d(i,j)之前,它保存的就是上一次刚刚算过的d(i-1,j)的值。同理,如果j要逆序枚举,d(i,j-w[i])保存的也是上次刚刚算过的d(i-1,j-w[i])的值。因此,忽略i,即可实现降维。

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define maxn 1000
#define rep(i,n) for(int i=0;i<(n);i++)
int d[maxn];
int w, v;
int vis[maxn];
int n, W;

int main()
{
	freopen("test.txt", "r", stdin);
	memset(d, 0, sizeof(d));//将d数组初始化为0
	scanf("%d%d", &n, &W);
	for (int i = 0; i < n; i++)//i顺序枚举,由递推公式决定
	{
		scanf("%d%d", &w, &v);//读入第i个物品的重量,价值
		for (int j = W; j >= 0;j--)
		if (j >= w)
			d[j] = max(d[j], d[j - w] + v);
	}
	printf("%d\n", d[W]);
	return 0;
}

上述代码也被称为“滚动数组”,正是因为它相当于在不断地刷新之前的结果。但这种方法容易产生bug,而且很难打印解。当动态规划结束后,只有最后一个阶段的值,而没有前面的值。因此这种方法应该谨慎使用。





动态规划小结——背包问题

标签:uva   动态规划   

原文地址:http://blog.csdn.net/u014800748/article/details/43760983

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!