CF286E Ladies' Shop

题意:

对于一个待求的集合{ai},用其对1-m中的所有数做完全背包,能装满n种大小的背包;

现给出n种背包{b1...bn}和m,求这个集合最小的大小并任意给出一种方案;


题解:

对于这个题最简单的集合就是{bi},而如果{bi}集合不合法的话,原问题就无解了;

判断这个集合不合法等价于集合中存在两个元素i,j且i+j∉该集合 (i+j<=m);

直接做是O(n^2)的,而可以发现将原b数组转成生成函数之后,这个问题就可以用卷积优化了;

判断好了有解,之后就是优化这个集合,让里面的元素最少了;

朴素的想法是从小到大枚举元素,然后判断这个元素能否被已经选中的元素表示,能则选中这个元素,否则不加入;

这个算法时间复杂度是O(n*m)的,每次都要做一遍完全背包;

而我们可以发现,判断这个元素是可选的不必这么麻烦,因为给出b数组的已经背包完成了,所以只需要判断当前数能否被两个i,j表示,一样是FFT解决;

时间复杂度O(mlogm);


代码:


#include<math.h>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 2097152
using namespace std;
const double pi=acos(-1.0);
struct cp
{
	double x,y;
	cp(){}
	cp(double _,double __):x(_),y(__){}
	friend cp operator +(cp a,cp b)
	{
		return cp(a.x+b.x,a.y+b.y);
	}
	friend cp operator -(cp a,cp b)
	{
		return cp(a.x-b.x,a.y-b.y);
	}
	friend cp operator *(cp a,cp b)
	{
		return cp(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
	}
}a[N],b[N];
void FFT(cp *a,int len,int type)
{
	int i,j,h,t;
	for(i=0,t=0;i<len;i++)
	{
		if(i>=t)	swap(a[i],a[t]);
		for(j=len>>1;(t^=j)<j;j>>=1);
	}
	for(h=2;h<=len;h<<=1)
	{
		cp wn(cos(2*pi*type/h),sin(2*pi*type/h));
		for(i=0;i<len;i+=h)
		{
			cp w(1,0),temp;
			for(j=0;j<h>>1;j++,w=w*wn)
			{
				temp=w*a[i+j+(h>>1)];
				a[i+j+(h>>1)]=a[i+j]-temp;
				a[i+j]=a[i+j]+temp;
			}
		}
	}
	if(type==-1)
		for(i=0;i<len;i++)
			a[i].x/=len;
}
int st[N],top;
bool A[N],B[N];
int main()
{
	int n,m,len,i,k;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&k);
		a[k].x=1;
		A[k]=1;
	}
	for(i=1<<25;i;i>>=1)
		if(m&i)
			{len=i<<2;break;}
	FFT(a,len,1);
	for(i=0;i<len;i++)
		b[i]=a[i]*a[i];
	FFT(b,len,-1);
	for(i=1;i<=m;i++)
	{
		if(!A[i]&&b[i].x>1e-5)
		{
			puts("NO");
			return 0;
		}
		if(A[i]&&b[i].x<1e-5)
		{
			st[++top]=i;
		}
	}
	puts("YES");
	printf("%d\n",top);
	for(i=1;i<=top;i++)
		printf("%d ",st[i]);
	return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值