首先上面的两道题都是同一道题,之所以同时拿出来是因为
Hdu 的数据比 Poj 的水,所以想区分一下。
Hdu 的数据对时间要求不高
但是 Poj 如果你想用简单的 多重背包解决的话,就会TLE 别问我怎么知道。
首先大家肯定会想到用多重背包来做
毕竟题意可以嵌套模板如下
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 100010;
int dp[maxn];
int c[maxn],w[maxn];
int v;
void zeroonepack(int cost,int wei) // 01
{
for(int i=v;i>=cost;i--){
dp[i] = max(dp[i],dp[i-cost] + wei);
}
}
void completepack(int cost,int wei) // 完全
{
for(int i=cost;i<=v;i++){
dp[i] = max(dp[i],dp[i-cost] + wei);
}
}
void multiplepack(int cost,int wei,int cnt) //多重
{
if(v < cnt*cost){
completepack(cost,wei);
return ;
}
else {
int k = 1;
while(k<=cnt)
{
zeroonepack(k*cost,k*wei);
cnt -= k;
k *= 2;
}
zeroonepack(cnt*cost,cnt*wei);
}
}
int main()
{
int n;
while(~scanf("%d%d",&n,&v)&& n+v){
for(int i=0;i<n;i++) scanf("%d",&c[i]);
for(int i=0;i<n;i++) scanf("%d",&w[i]);
memset(dp,0,sizeof(dp));
for(int i=0;i<n;i++) multiplepack(c[i],c[i],w[i]);
int sum = 0;
for(int i=1;i<=v;i++) if(dp[i]==i) sum++;//因为代价和价值是相同的
printf("%d\n",sum);
}
return 0;
}
额,代码是对的,答案也是对的,但是你会发现他超时了,因为复杂度是O(V*Σn[i]) ,所以太大,Hdu 和 Poj 都交不对。
所以我们就想优化一下,于是按照老思路,可不可以减少多重背包对数量的循环次数呢?
我们把一个物品根据数量把它变成多个物品,怎么变会少一些时间呢?
答案是根据二进制来变。
数量为num的物品,我们把它分成系数为1,2,4,8 ···· 2^n (2^n<= num)的话,选的时候花费和价值都乘以这个系数是不是就可以把它优化的分为几个物品。
并且显然满足num = 1+2+4+8+2^n (2^n<= num)。
这样的话时间就变成O(V*Σlog n[i]) 的01背包了。
代码如下
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 100010;
int dp[maxn];
int c[maxn],w[maxn];
int v;
int main()
{
int n;
while(~scanf("%d%d",&n,&v)&& n+v){
for(int i=0;i<n;i++) scanf("%d",&c[i]);
for(int i=0;i<n;i++) scanf("%d",&w[i]);
memset(dp,0,sizeof(dp));
dp[0] = 1;
for(int i=0;i<n;i++){
int cnt = w[i];
for(int k=1;k<=cnt;k<<=1){ //二进制划分
for(int j=v;j>=k*c[i];j--){
dp[j] += dp[j-k*c[i]];
}
cnt -= k;
}
if(cnt){
for(int j=v;j>=cnt*c[i];j--) dp[j] += dp[j-cnt*c[i]];
}
}
int sum = 0;
for(int i=1;i<=v;i++) if(dp[i]) sum++;
printf("%d\n",sum);
}
return 0;
}
然而,这样子的代码只能过了Hdu的数据,对于Poj上的数据,还是超时。
现在怎么办,我根据网上其他人的代码看出了怎么优化。
第一个优化就是,再开一个数组来记录某个数用了多少次,dp数组就直接用来标记这个数有没有组成过,这个优化我觉得比较很重要
第二个优化就是计数答案就直接在循环里面记录
代码如下
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 100010;
int dp[maxn];
int sum[maxn];
int c[maxn],w[maxn];
int v;
int main()
{
int n;
while(~scanf("%d%d",&n,&v)&& n+v){
for(int i=0;i<n;i++) scanf("%d",&c[i]);
for(int i=0;i<n;i++) scanf("%d",&w[i]);
memset(dp,0,sizeof(dp));
dp[0] = 1;
int ans = 0;
for(int i=0;i<n;i++){
memset(sum,0,sizeof(sum));
for(int j=c[i];j<=v;j++){
if(!dp[j] && dp[j-c[i]] && sum[j-c[i]] < w[i]){
dp[j] = 1;
sum[j] = sum[j-c[i]] +1;
ans++;
}
}
}
printf("%d\n",ans);
}
return 0;
}
因为本人比较菜,右移代码有所参考网络上的,如有侵犯,私聊我会改。