0-1背包问题

递归暴力搜索
#include <iostream>
#define MAX 10000
using namespace std; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
typedef pair<int,int> p;
int n,W;
p A[MAX];
int ret(int i,int j);
int main(int argc, char** argv) 
{
	cin>>n>>W;
	for(int i=0;i<n;i++)
	{
		int w,v;
		cin>>w>>v;
		A[i].first=w;
		A[i].second=v;
	}
	cout<<ret(0,W);
	return 0;
}
int ret(int i,int j)
{
	int res;
	if(i==n)
	{
		res=0; 
	}
	else if(j<A[i].first) 
	{
		res=ret(i+1,j);
	}
	else
	{
		res=max(ret(i+1,j),ret(i+1,j-A[i].first)+A[i].second);
	}
	return res;
	
} 

通过记忆化搜索可以将复杂度减到O(n*W)
#include <iostream>
#define MAX 10000
using namespace std; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
typedef pair<int,int> p;
int n,W;
p A[MAX];
int ret(int i,int j);
int dp[MAX][MAX];
int main(int argc, char** argv) 
{
	cin>>n>>W;
	for(int i=0;i<n;i++)
	{
		int w,v;
		cin>>w>>v;
		A[i].first=w;
		A[i].second=v;
	}
	cout<<ret(0,W);
	return 0;
}
int ret(int i,int j)
{
	if(dp[i][j]>0)
	{
		return dp[i][j];
	}
	int res;
	if(i==n)
	{
		res=0; 
	}
	else if(j<A[i].first) 
	{
		res=ret(i+1,j);
	}
	else
	{
		res=max(ret(i+1,j),ret(i+1,j-A[i].first)+A[i].second);
	}
	return dp[i][j]=res;
	
} 

dp[i][j]:=从第i个数开始选,还能装的重量为j
dp[i][j]=max(dp[i+1][j],dp[i+1][j-w[i]+v[i])
#include <iostream>
#define MAX 10000
using namespace std; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
typedef pair<int,int> p;
int n,W;
p A[MAX];
int ret(int i,int j);
int dp[MAX][MAX];
void solve();
int main(int argc, char** argv) 
{
	cin>>n>>W;
	for(int i=0;i<n;i++)
	{
		int w,v;
		cin>>w>>v;
		A[i].first=w;
		A[i].second=v;
	}
	solve();
	cout<<dp[0][W];
	return 0;
}
void solve()
{
	for(int i=n-1;i>=0;i--)
	{
		for(int j=0;j<=W;j++)
		{
			if(j<A[i].first)
			{
				dp[i][j]=dp[i+1][j];
			}
			else
			{
				dp[i][j]=max(dp[i+1][j],dp[i+1][j-A[i].first]+A[i].second);
			}
		}
	}
}

阅读更多
个人分类: dynamic programming
上一篇413. Arithmetic Slices
下一篇堆的实现
想对作者说点什么? 我来说一句

算法设计0-1背包问题

2010年04月10日 542B 下载

0-1背包问题贪心算法源码下载

2009年06月29日 24KB 下载

没有更多推荐了,返回首页

关闭
关闭