POJ3624 01背包入门题[DP]

01背包第一题。

刚开始开的数组为int mat[N][W]，然后内存超了。

然后优化成这样既可。

#include<iostream>
#define max(a,b) (a>b?a:b)
using namespace std;
//const int N=3410;
const int W=13888;
int mat[2][W];
int n,w;
int main()
{
	scanf("%d%d",&n,&w);
	int wi,di;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&wi,&di);
		for(int j=1;j<wi;j++)
		{
			mat[i%2][j]=mat[(i-1)%2][j];
		}
		for(int j=wi;j<=w;j++)
		{
			mat[i%2][j]=max(mat[(i-1)%2][j],mat[(i-1)%2][j-wi]+di);
		}
	}
	printf("%d\n",mat[n%2][w]);
	return 0;
}

然后再优化，优化成1维的如下：

#include<iostream>
#define max(a,b) (a>b?a:b)
using namespace std;
//const int N=3410;
const int W=13888;
int mat[W];
int n,w;
int main()
{
	
	scanf("%d%d",&n,&w);
	int wi,di;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&wi,&di);
		for(int j=w;j>=1;j--)
		{
			if(j>=wi)
				mat[j]=max(mat[j],mat[j-wi]+di);
		}
	}
	printf("%d\n",mat[w]);
	return 0;
}