01揹包问题动态规划算法

#include <cstdio>
#define max(u,v) (u)>(v)?(u):(v)
#define min(u,v) (u)>(v)?(v):(u)

int m[10][20];
int v[10];
int w[10];
int x[10];
void Knapsack(int v[10], int w[10], int c, int n, int m[10][20])
{
	int i,j,jMax;
	jMax=min(w[n]-1,c);
	for(j=0;j<=jMax;j++) m[n][j]=0;
	for(j=jMax+1;j<=c;j++) m[n][j]=v[n];
	for(i=n-1;i>1;i--)
	{
		jMax=min(w[i]-1,c);
		for(j=0;j<=jMax;j++)
			m[i][j]=m[i+1][j];
		for(j=jMax+1;j<=c;j++)
			m[i][j]=max(m[i+1][j],m[i+1][j-w[i]]+v[i]);		
	}
	m[1][c]=m[2][c];
	if(w[1]<=c)
		m[1][c]=max(m[1][c],m[2][c-w[1]]+v[1]);
}

void Traceback(int m[10][20], int w[10], int c, int n, int x[10])
{
	int i;
	for(i=1;i<n;i++)
		if(m[i][c]==m[i+1][c])
			x[i]=0;
		else
		{
			x[i]=1;
			c-=w[i];
		}
		x[n]=(m[n][c])?1:0;
}
 
int main()
{
	int i,c=10,n=5;
	w[1]=2;w[2]=2;w[3]=6;w[4]=5;w[5]=4;
	v[1]=6;v[2]=3;v[3]=5;v[4]=4;v[5]=6;
	Knapsack(v,w,c,n,m);
	Traceback(m,w,c,n,x);
	for(i=1;i<=n;i++)
		if(x[i])
			printf("(w=%d,v=%d)\n",w[i],v[i]);
	return 0;
}

点赞