有一个背包,体积为 m m m。现在给出 n n n种物品,每种物品体积为 a i a_i ai,价值为 b i b_i bi,个数为 c i c_i ci。求这个背包能装下物品的最大价值为多少。
如果把第 i i i种物品拆成 c i c_i ci个物品来用01背包的方法做,则时间复杂度为 O ( m ∑ c i ) O(m\sum c_i) O(m∑ci),在一些题目中会TLE。所以我们需要用到二进制优化。
对于题中的 c i c_i ci,我们可以把它分为 ( 2 t − 1 ) + ( c i − 2 t + 1 ) (2^t-1)+(c_i-2^t+1) (2t−1)+(ci−2t+1),其中 t t t是满足 c i − 2 t + 1 ≥ 0 c_i-2^t+1\geq 0 ci−2t+1≥0的最大整数。
2 t − 1 2^t-1 2t−1可以分为 1 , 2 , 4 , … , 2 t − 2 , 2 t − 1 1,2,4,\dots,2^{t-2},2^{t-1} 1,2,4,…,2t−2,2t−1,这些数可以组成 1 1 1到 2 t − 1 2^t-1 2t−1中的所有数。
于是,我们可以将第 i i i种物品分成 t + 1 t+1 t+1组,第 j j j组物品 ( 1 ≤ j ≤ t ) (1\leq j\leq t) (1≤j≤t)有 2 j − 1 2^{j-1} 2j−1个,第 t + 1 t+1 t+1组物品有 ( c i − 2 t + 1 ) (c_i-2^t+1) (ci−2t+1) 个
假设第 i i i种物品需要放 k k k个
若 1 ≤ k ≤ 2 t − 1 1\leq k\leq 2^t-1 1≤k≤2t−1,显然以上的数据可以组成。
若
2
t
≤
k
≤
c
i
2^t\leq k\leq c_i
2t≤k≤ci
2
t
−
(
c
i
−
2
t
+
1
)
≤
k
−
(
c
i
−
2
t
+
1
)
≤
c
i
−
(
c
i
−
2
t
+
1
)
2^t-(c_i-2^t+1)\leq k-(c_i-2^t+1)\leq c_i-(c_i-2^t+1)
2t−(ci−2t+1)≤k−(ci−2t+1)≤ci−(ci−2t+1)
也就是
2
t
+
1
−
c
i
−
1
≤
k
−
(
c
i
−
2
t
+
1
)
≤
2
t
−
1
2^{t+1}-c_i-1\leq k-(c_i-2^t+1)\leq 2^t-1
2t+1−ci−1≤k−(ci−2t+1)≤2t−1。
我们只需要用第
1
1
1到
t
t
t组物品凑出
k
−
(
c
i
−
2
t
+
1
)
k-(c_i-2^t+1)
k−(ci−2t+1)个,再加上第
t
+
1
t+1
t+1组物品的
(
c
i
−
2
t
+
1
)
(c_i-2^t+1)
(ci−2t+1)个,就得出
k
k
k个物品了。
用二进制可以很好地提高代码的效率,总时间复杂度为
O
(
m
∑
l
o
g
c
i
)
O(m\sum log c_i)
O(m∑logci)。
code
#include<bits/stdc++.h>
using namespace std;
int n,m,a[10005],b[10005],c[10005],f[10005];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d%d",&a[i],&b[i],&c[i]);
}
for(int i=1;i<=n;i++){
int v=1,p=a[i],q=b[i];
while(c[i]>=v){
c[i]-=v;
for(int j=m;j>=p;j--){
f[j]=max(f[j],f[j-p]+q);
}
v<<=1;
p=a[i]*v;q=b[i]*v;
}
if(c[i]){
p=a[i]*c[i];q=b[i]*c[i];
for(int j=m;j>=p;j--){
f[j]=max(f[j],f[j-p]+q);
}
}
}
printf("%d",f[m]);
return 0;
}
例题HDU-1171 Big Event in HDU
可以用二进制优化
code
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,sum,a,b,f[1000005];
int main()
{
while(scanf("%d",&n)){
if(n<0) return 0;
memset(f,0,sizeof(f));
f[0]=1;sum=0;
for(int i=1;i<=n;i++){
scanf("%d%d",&a,&b);
sum+=a*b;
int v=1,p=a;
while(b>=v){
b-=v;
for(int j=sum;j>=p;j--) f[j]|=f[j-p];
v*=2;p=a*v;
}
if(b){
p=a*b;
for(int j=sum;j>=p;j--) f[j]|=f[j-p];
}
}
for(int i=(sum+1)/2;;i++){
if(f[i]){
printf("%d %d\n",i,sum-i);
break;
}
}
}
}