背包问题总结

学习于：背包九讲

01背包

P1048 [NOIP2005 普及组] 采药
基础01背包，一个限制条件，每个物品只能用一次，求价值最大。

#include<stdio.h>
#include<algorithm>
using namespace std;
int w[300],v[300],dp[100000],n,m;
int main()
{
	int i,j;
	scanf("%d %d",&m,&n);
	for(i=1;i<=n;i++)
		scanf("%d %d",&w[i],&v[i]);
	for(i=1;i<=n;i++)
		for(j=m;j>=w[i];j--)
			dp[j]=max(dp[j-w[i]]+v[i],dp[j]);	
	printf("%d",dp[m]);
	return 0;
}

完全背包

P1616 疯狂的采药
完全背包，一个限制条件，每个物品无限次用，求价值最大。

#include<stdio.h>
#include<algorithm>
using namespace std;
int w[10005],v[10005],dp[100005],n,m;
int main()
{
	int i,j;
	scanf("%d %d",&m,&n);
	for(i=1;i<=n;i++)
		scanf("%d %d",&w[i],&v[i]);
	for(i=1;i<=n;i++)
		for(j=w[i];j<=m;j++)
			dp[j]=max(dp[j-w[i]]+v[i],dp[j]);	
	printf("%d",dp[m]);
	return 0;
}

另：P1853 投资的最大效益

多重背包

Luogu P1776 宝物筛选
多重背包是01背包加完全背包的问题，限制条件内每种物品数量有限，求价值最大。

基础方法

p[i]:每个物品的件数，0代表无穷个
for (int i = 1; i <= n; i++)
    if (p[i] == 0)
        for (int j = w[i]; j <= V; j++)
            f[j] = max(f[j], f[j - w[i]] + v[i]);
    else
    for (int k = 1; k <= p[i]; k++)
        for (int j = V; j >= w[i]; j--)
            f[j] = max(f[j], f[j - w[i]] + v[i]);

二进制分解求解

#include<stdio.h>
#include<algorithm>
#include<cstring>
using namespace std;
int w[1000005],v[1000005],s[1000005],dp[10000005],n,m;
int main()
{
	int i,j,k,t;
	scanf("%d %d",&n,&m);
	for(i=1;i<=n;i++)
		scanf("%d %d %d",&v[i],&w[i],&s[i]);
	for(i=1;i<=n;i++){
		int num=min(s[i],m/w[i]);
		for(k=1;num>0;k<<=1){			
			if(k>num) k=num;
			num-=k;
			for(j=m;j>=w[i]*k;j--)
				dp[j]=max(dp[j-w[i]*k]+v[i]*k,dp[j]);
		}
	}
	printf("%d",dp[m]);
	return 0;
}

二维费用背包

P1507 NASA的食物计划
在基础01背包上多加一个限制条件，求价值最大。

#include<stdio.h>
#define max(x,y) x>y?x:y
int main()
{
	int x1,y1,v,i,j,k,n,a[505][505],x,y;
	scanf("%d %d %d",&x,&y,&n);
	for(i=1;i<=n;i++){
		scanf("%d %d %d",&x1,&y1,&v);
		for(j=x;j>=x1;j--)
			for(k=y;k>=y1;k--)
				a[j][k]=max(a[j-x1][k-y1]+v,a[j][k]);				
	}
	printf("%d",a[x][y]);
	return 0;
}

分组背包

P1757 通天之分组背包
物品分组，每组只能用一个。

for (所有的组k)
    for (int j = V; j >= 0; j--)
        for (所有属于组k的i)
            f[j] = max{f[j], f[j - w[i]] + v[i]}

#include<stdio.h>
#define max(x,y) x>y?x:y
#define min(x,y) x<y?x:y
int a[105][105],w[1005],v[1005],c[1005],dp[1005];
int main()
{
	int x1,x2,y1,y2,i,j,k,u,n,m,s,x,y,mc=0;
	scanf("%d %d",&m,&n);
	for(i=1;i<=n;i++){
		scanf("%d %d %d",&w[i],&v[i],&k);
		mc=max(k,mc);
		c[k]++;
		a[k][c[k]]=i;
	}
	for(i=1;i<=mc;i++)
		for(j=m;j>=0;j--)
			for(k=1;k<=c[i];k++)
				if(j>=w[a[i][k]])
					dp[j]=max(dp[j-w[a[i][k]]]+v[a[i][k]],dp[j]);
	printf("%d",dp[m]);
	return 0;
}

有依赖背包问题

P1064 [NOIP2006 提高组] 金明的预算方案
选择物品有前置条件，必须有一个物品最先选。

#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
struct node{
	int w,v;
};
vector<node>v[100];
int fw[105][5],fv[105][5],w[105],c[105],dp[200005];
int main()
{
	int n,m,i,j,k,x,y;
	scanf("%d %d",&m,&n);
	for(i=1;i<=n;i++){
		scanf("%d %d %d",&x,&y,&k);
		if(!k) v[i].push_back((node){x,x*y});
		else {
			for(j=0;j<v[k].size();j++)
				v[k].push_back((node){x+v[k][j].w,x*y+v[k][j].v});
		}
	}
	for(i=1;i<=n;i++)
		for(j=m;j>=0;j--)
			for(k=0;k<v[i].size();k++){
			node temp=v[i][k];
			if(j>=temp.w)
				dp[j]=max(dp[j-temp.w]+temp.v,dp[j]);
		}
	printf("%d",dp[m]);
	return 0;
}

泛化背包

Bone Collector II
第k优解

#include<stdio.h>
#include<string.h>
#define max(x,y) x>y?x:y
#define min(x,y) x<y?x:y
int w[105],v[105],c[105],dp[2005][50],a[50],b[50];
int main()
{
    int n,m,i,j,k,x,y,t,u;
    scanf("%d",&t);
    while(t--){
        scanf("%d %d %d",&n,&m,&k);
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++) 
            scanf("%d",&v[i]);
        for(i=1;i<=n;i++) 
            scanf("%d",&w[i]);    
        for(i=1;i<=n;i++){
            for(j=m;j>=w[i];j--){
                for(u=1;u<=k;u++){
                    a[u]=dp[j][u];
                    b[u]=dp[j-w[i]][u]+v[i];
                }    
                a[k+1]=-1;
                b[k+1]=-1;
                u=x=y=1;
                while(u!=k+1&&(a[x]!=-1||b[y]!=-1)){
                    if(a[x]>b[y])
                        dp[j][u]=a[x++];
                    else 
                        dp[j][u]=b[y++];
                    if(dp[j][u]!=dp[j][u-1]) u++;
                }
                
            }
        }
        printf("%d\n",dp[m][k]);
    }
    return 0;
}