vijos搭建双塔(dp)

数据弱,朴素的算法竟然过了,后来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
#include
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
#include
#include
#include
#include
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]<0;ans--) if (f[ans].test(ans+1)) break;
	if (ans) printf("%d",ans);else printf("Impossible");
	return 0;
}


你可能感兴趣的:(dp)