您的位置：首页 > 其它

POJ3624 01背包入门题[DP]

2011-10-31 23:16 232 查看

01背包第一题。

刚开始开的数组为int mat
[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;
}

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航