01背包模板

一、二维写法

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 1010;

int v[N],w[N];
int f[N][N];
int n,m;

int main()
{
    cin>>n>>m;

    for(int i=1;i<=n;i++) cin>>v[i]>>w[i];

    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            f[i][j] = f[i-1][j];    // 不选第 i 个物品
            if(j>=v[i])             // 选第 i 个物品
                f[i][j] = max(f[i][j],f[i-1][j-v[i]]+w[i]);
        }
    }

    int res=0;
    for(int i=1;i<=m;i++) res = max(res,f[n][i]);

    cout<<res<<endl;

    // 或者直接输出 f[n][m];
    // cout<<f[n][m]<<endl;

    return 0;
}

二、一维写法

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 1010;
int n,m;
int v[N],w[N];
int f[N];

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>v[i]>>w[i];

    for(int i=1;i<=n;i++)
    {
        for(int j=m;j>=v[i];j--)
        {
            f[j]=max(f[j],f[j-v[i]]+w[i]);
        }
    }

    cout<<f[m]<<endl;

    return 0;
}

01背包模板


版权声明:本文为qq_45856289原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。