您的位置:首页 > 其它

vijos搭建双塔(dp)

2016-05-28 17:10 218 查看
数据弱,朴素的算法竟然过了,后来bitset位运算的01背包优化加上后快了7倍//顺便还练习了一下对拍

f【i】【j】=true,表示塔1高为i,塔2高为j,的情况存在

方程

if (dp[j][l])
{
dp[j+a[i]][l]=true;
dp[j][l+a[i]]=true;
}


朴素:

#include<cstdio>
#include<algorithm>
using namespace std;
int a[105],n;
bool dp[5000][5000];
int main()
{
int n;
scanf("%d",&n);
int sum=0;
for (int i=1;i<=n;i++) scanf("%d",&a[i]),sum+=a[i];
dp[0][0]=true;
for (int i=1;i<=n;i++)
{
for (int j=sum;j>=0;j--)
for (int l=sum;l>=0;l--)
if (dp[j][l]) { dp[j+a[i]][l]=true; dp[j][l+a[i]]=true; }
}
int ans;
for(ans=sum;ans>0;ans--)if (dp[ans][ans]) break;
if (ans)printf("%d",ans);else printf("Impossible");
return 0;
}


bitset优化的

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<bitset>
using namespace std;
int w[105],n,h[105],sum=0;

int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)scanf("%d",&w[i]),sum+=w[i];
bitset<3000> f[3000];
for (int i=0;i<=sum;i++) f[i].reset();
f[0].set(1);
for (int i=1;i<=n;i++)
{
for (int j=sum;j>=0;j--) if (f[j].any())
{
f[j+w[i]]=f[j]|f[j+w[i]];//位运算的精髓
f[j]=f[j]|(f[j]<<w[i]);//精髓!
}
}
int ans;
for (ans=sum;ans>0;ans--) if (f[ans].test(ans+1)) break;
if (ans) printf("%d",ans);else printf("Impossible");
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: