背包问题

01背包

二维：

#include<iostream>
using namespace std;
int dp[100][100];
int w[100],v[100],c[100];
int n,W;
int main()
{
    while(cin>>n)
    {
        for(int i=1;i<=n;i++)
        cin>>w[i];
        for(int i=1;i<=n;i++)
        cin>>v[i];
        cin>>W;
        for(int i=0;i<=n;i++)
        dp[i][0]=0;
        for(int j=0;j<=W;j++)
        dp[0][j]=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=W;j++)
            {
                if(j<w[i])
                dp[i][j]=dp[i-1][j];
                else
                {
                    if(dp[i-1][j]>dp[i-1][j-w[i]]+v[i])
                    dp[i][j]=dp[i-1][j];
                    else
                    dp[i][j]=dp[i-1][j-w[i]]+v[i];
                }
            }
        }
        cout<<"最大价值";
        cout<<dp[n][W]<<endl;
        int num=0;
        int max=0;
        int r=W;
        for(int i=n;i>0;i--)
        {
            if(dp[i][r]!=dp[i-1][r])
            {
                c[++num]=i;
                max=max+w[i];
                r=r-w[i];
            }
        }
        cout<<"最大重量"<<max<<endl;
        cout<<"选测情况:"<<endl;
        for(int i=num;i>=1;i--)
        cout<<c[i]<<" ";
        cout<<endl;
    }
    return 0;
}

一维：

#include<iostream>
#include<string.h>
using namespace std;
int t;
int n,M;
int v[1010];
int m[1010];
int dp[1010];
int main()
{
    cin>>t;
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        cin>>n>>M;
        for(int i=1;i<=n;i++)
            cin>>v[i];
        for(int i=1;i<=n;i++)
            cin>>m[i];
        for(int i=1;i<=n;i++)
        {
            for(int j=M;j>=m[i];j--)
            {
                dp[j]=max(dp[j],dp[j-m[i]]+v[i]);
            }
        }
        cout<<dp[M]<<endl;
    }
    return 0;
}

完全背包：

#include<iostream>
#include<string.h>
using namespace std;
int t;
int n,M;
int v[1010];
int m[1010];
int dp[1010];
int main()
{
    cin>>t;
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        cin>>n>>M;
        for(int i=1;i<=n;i++)
            cin>>v[i];
        for(int i=1;i<=n;i++)
            cin>>m[i];
        for(int i=1;i<=n;i++)
        {
            for(int j=m[i];j<=M;j++)
            {
                dp[j]=max(dp[j],dp[j-m[i]]+v[i]);
            }
        }
        cout<<dp[M]<<endl;
    }
    return 0;
}

多重背包

#include<iostream>
#include<string.h>
using namespace std;
int t,n,m,p[110],h[110],c[110],dp[110];
int main()
{
    cin>>t;
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        cin>>n>>m;
        for(int i=1;i<=m;i++)
        cin>>p[i]>>h[i]>>c[i];
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=c[i];j++)
            {
                for(int k=n;k>=p[i];k--)
                {
                    dp[k]=max(dp[k],dp[k-p[i]]+h[i]);
                }
            }
        }
        cout<<dp[n]<<endl;
    }
    return 0;
}