模板题——背包问题

1.01背包

思路分析

状态表示：f[i][j]表示只看前i个物品，总体积是j的情况下，总价值最大是多少
答案是：result=max{f[n][0~v]}
状态转移：f[i][j]
1.不选第i个 f[i][j]=f[i-1][j]
2.选第i个 f[i][j]=f[i-1][j-v[i]]
初始化：f[0][0]=0
f[i][j]=max(1.2.)

朴素代码

#include <bits/stdc++.h>

using namespace std;
const int N=1010;
int n,m;
int f[N][N];//全局变量已经初始化完了f[0][0]=0
int v[N],w[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=0;j<=m;j++)
    {
        f[i][j]=f[i-1][j];
        if(j>=v[i])
            f[i][j]=max(f[i][j],f[i-1][j-v[i]]+w[i]);
    }
    cout<<f[n][m]<<endl;
    return 0;
}

代码优化【按照朴素算法等价优化空间复杂度，单纯优化代码逻辑，与题目无关】

#include <bits/stdc++.h>

using namespace std;
const int N=1010;
int n,m;
int f[N];
int v[N],w[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;
}

2.完全背包

完全背包的状态转移和01背包的区别
01背包：f[i][j]=max(f[i-1][j],f[i-1][j-v]+w)
完全背包：f[i][j]=max(f[i-1][j],f[i][j-v]+w)

朴素做法

#include <bits/stdc++.h>

using namespace std;
const int N=1010;
int n,m;
int f[N][N],w[N],v[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=1;j<=m;j++)
    {
        f[i][j]=f[i-1][j];
        if(j>=v[i]) f[i][j]=max(f[i][j],f[i][j-v[i]]+w[i]);
    }
    cout<<f[n][m]<<endl;
    return 0;
}

空间优化版【玄学？No！科学！！】

#include <bits/stdc++.h>

using namespace std;
const int N=1010;
int n,m;
int f[N],w[N],v[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=v[i];j<=m;j++)
            f[j]=max(f[j],f[j-v[i]]+w[i]);
    cout<<f[m]<<endl;
    return 0;
}

3.多重背包

数据很少的版本：三重循环

#include <bits/stdc++.h>

using namespace std;
const int N=110;
int n,m;
int v[N],w[N],s[N];
int f[N][N];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>v[i]>>w[i]>>s[i];
    for(int i=1;i<=n;i++)
        for(int j=0;j<=m;j++)
            for(int k=0;k<=s[i]&&k*v[i]<=j;k++)
                f[i][j]=max(f[i][j],f[i-1][j-v[i]*k]+w[i]*k);
    cout<<f[n][m]<<endl;
    return 0;
}

数据很多的二进制优化版本

#include <bits/stdc++.h>

using namespace std;
const int N=12010,M=2010;
int n,m;
int v[N],w[N];
int f[M];
int main()
{
    cin>>n>>m;
    int cnt=0;//将s[i]个i物品总和按照二进制拆分，并用新的物品组成01背包
    for(int i=1;i<=n;i++)
    {
        int a,b,s;
        cin>>a>>b>>s;
        int k=1;
        while(k<=s)//将k个打包为一个
        {
            cnt++;
           v[cnt]=a*k;
            w[cnt]=b*k;
            s-=k;k*=2;
        }
        if(s>0)
        {
            cnt++;
            v[cnt]=a*s;
            w[cnt]=b*s;
        }
    }
    n=cnt;//重新打包后的背包总数
    //01背包模板
    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;
}

4.分组背包

思路：先枚举每一个组，再在组内选择
for(int i=1;i<=n;i++)
{  for(int j=m;j>=0;j--)
       f[j]=max(f[j],f[j-v[0]]+w[0],f[j-v[1]]+w[i],...f[j-v[s-1]]+w[s-1])
 }

 多重背包和分组背包的区别：
 多重背包：枚举第i个背包选几个
 分组背包：枚举第i组物品选哪个

#include <bits/stdc++.h>

using namespace std;
const int N=110;
int n,m;
int v[N],f[N],w[N];
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        int s;
        cin>>s;
        for(int j=0;j<s;j++) cin>>v[j]>>w[j];
        for(int j=m;j>=0;j--)
            for(int k=0;k<s;k++)
            if(j>=v[k])
            f[j]=max(f[j],f[j-v[k]]+w[k]);
    }
    cout<<f[m]<<endl;
    return 0;
}