数据弱,朴素的算法竟然过了,后来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;
}