题目大意:
n 只奶牛坐电梯,每只奶牛重ci,电梯最大承重为W,问至少乘坐多少
次电梯才能使所有奶牛离开。每次乘坐时,电梯里奶牛总重不能超过电
梯最大承重
n<=20,1<=ci,w<=1e9
solution:
注意到n 只有20,考虑使用状压DP
f(S) 表示集合S 中的奶牛至少需要乘几次电梯
预处理集合S 中奶牛总重是否超过W,然后通过枚举子集的技巧进行转移,时间复杂度O(3^n)
瓶颈在于子集DP 时间复杂度始终为O(3n) 无法优化
考虑不枚举子集,而枚举新加入一个奶牛。这时决策需要知道当前电梯是否能装下新奶牛
记录辅助数组g(S) 表示在f(S) 最优时,最后一个电梯剩余的最大承重
时间复杂度O(n*2^n)
这里我们要注意到枚举子集的方法复杂度都是3^n级别的
这种1s内只能跑过n<=16的,17都不行
所以如果不枚举子集可以枚举新加入的,利用辅助数组记录额外的信息
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 20
using namespace std;
int n,w,a[maxn],f[1<<maxn],g[1<<maxn];
int main(){
scanf("%d%d",&n,&w);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
memset(f,0x3f,sizeof f);
g[0]=w; f[0]=1;
for(int i=0;i<(1<<n);i++)
for(int j=1;j<=n;j++){
if(i&(1<<(j-1))) continue;
if(g[i]>=a[j] && f[i|(1<<(j-1))]>=f[i]){//这里注意if里的条件
f[i|(1<<(j-1))]=f[i];
if(f[i|(1<<(j-1))]==f[i]) g[i|(1<<(j-1))]=max(g[i]-a[j],g[i|(1<<(j-1))]);
else g[i|(1<<(j-1))]=g[i]-a[j];//还有这里
}
else if(g[i]<a[j] && f[i|(1<<(j-1))]>=f[i]+1){
f[i|(1<<(j-1))]=f[i]+1;
if(f[i|(1<<(j-1))]==f[i]+1) g[i|(1<<(j-1))]=max(w-a[j],g[i|(1<<(j-1))]);
else g[i|(1<<(j-1))]=w-a[j];
}
}
printf("%d\n",f[(1<<n)-1]);
return 0;
}
本文介绍了一个经典的状压动态规划问题——如何计算最少电梯次数来运送一群特定重量的奶牛。通过枚举新加入的奶牛并利用辅助数组记录额外信息的方式,优化了算法的时间复杂度。

1302

被折叠的 条评论
为什么被折叠?



