Dividing 背包

Dividing
原题链接https://vjudge.net/contest/348156#problem/L
在这里插入图片描述
在这里插入图片描述
题目中给出1-6号编号的弹珠的数量 要求判断是否可以平分
计算是否有总数的一半这种情况
由于数据较大我们采用二进制优化来写,
二进制优化:1 2 4 8 。。。可以组成1-所有的数,即可枚举所有情况

#include<cstring>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;
long long mv[6]= {1,2,3,4,5,6};
long long num[6];
long long dp[100005];
int main()
{
	long long z,i,j,sum2,lala=0;
	while(~scanf("%lld %lld %lld %lld %lld %lld",&num[0],&num[1],&num[2],&num[3],&num[4],&num[5]))
	{
		if(num[0]==0&&num[1]==0&&num[2]==0&&num[3]==0&&num[4]==0&&num[5]==0)
		{
			break;
		}
		memset(dp,-1,sizeof(dp));
		long long sum=0;
		for(i=0; i<6; i++)
		{
			sum+=mv[i]*num[i];
		}
		lala++;
		if(sum%2==1)//如果总数是奇数即不可平分
		{
			printf("Collection #%lld:\n",lala);
			printf("Can't be divided.\n\n");
			continue;
		}
		long long sum1=sum/2;
		dp[0]=1;
		for(i=0; i<6; i++)
		{
			if(num[i]==0)
			{
				continue;
			}
			for(j=1; j<=num[i]; j=j*2)//二进制优化
			{
				sum2=mv[i]*j;
				for(z=sum1; z>=sum2; z--)
				{
					if(dp[z-sum2]==1)
					{
						dp[z]=1;
					}
				}
				num[i]-=j;
			}
			sum2=num[i]*mv[i];
			if(sum2>0)//对二进制后还剩下的数进行计算
			{
				for(z=sum1; z>=sum2; z--)
				{
					if(dp[z-sum2]==1)
					{
						dp[z]=1;
					}
				}
			}
		}
		if(dp[sum1]==1)//判断这种情况是否成立
		{
			printf("Collection #%lld:\n",lala);
			printf("Can be divided.\n\n");
		}
		else
		{
			printf("Collection #%lld:\n",lala);
			printf("Can't be divided.\n\n");
		}
	}
	return 0;
}

发布了130 篇原创文章 · 获赞 3 · 访问量 1627

猜你喜欢

转载自blog.csdn.net/yeyuluo/article/details/103896849