题意:
对于一个待求的集合{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;
}