CF286E Ladies' Shop FFT
读完题后,我们发现如下性质:
- 在合法且和不超过 $m$ 的情况下,如果 $a_{i}$ 出现,则 $a_{i}$ 的倍数也必出现.
- 所以如果合法,只要对所有数两两结合一次就能得到所有 $a_{i}$ 能凑出的不超过 $m$ 的数,且没有多余的.
- 只需对 $a_{1}...a_{n}$ 两两结合,如果发现一个新的数被凑出来,而这个数在 $a_{1}...a_{n}$ 没出现过,那么就输出无解,否则,就找出那些不能被两两结合出来的 $a_{i}$,输出即可
#include<bits/stdc++.h> #define setIO(s) freopen(s".in","r",stdin) #define maxn 4000000 #define ll long long using namespace std; const double pi=acos(-1.0); struct cpx { double x,y; cpx(double a=0,double b=0) {x=a,y=b; } cpx operator+(const cpx b) { return cpx(x+b.x,y+b.y); } cpx operator-(const cpx b) { return cpx(x-b.x,y-b.y); } cpx operator*(const cpx b) { return cpx(x*b.x-y*b.y,x*b.y+y*b.x); } }A[maxn]; void FFT(cpx *a,int n,int flag) { for(int i=0,k=0;i<n;++i) { if(i>k) swap(a[i],a[k]); for(int j=n>>1;(k^=j)<j;j>>=1); } for(int mid=1;mid<n;mid<<=1) { cpx wn(cos(pi/mid), flag*sin(pi/mid)),x,y; for(int i=0;i<n;i+=(mid<<1)) { cpx w(1,0); for(int j=0;j<mid;++j) { x=a[i+j],y=w*a[i+j+mid]; a[i+j]=x+y,a[i+j+mid]=x-y; w=w*wn; } } } if(flag==-1) for(int i=0;i<n;++i) a[i].x/=(double)n; } int ans[maxn],arr[maxn],bu[maxn]; int main() { // setIO("input"); int n,m,len; scanf("%d%d",&n,&m); for(int i=1;i<=n;++i) scanf("%d",&arr[i]); for(int i=1;i<=n;++i) bu[arr[i]]++,ans[arr[i]]++; for(int i=1;i<=m;++i) if(bu[i]) A[i].x=1; for(len=1;len<=(m<<1);len<<=1); FFT(A,len,1); for(int i=0;i<len;++i) A[i]=A[i]*A[i]; FFT(A,len,-1); for(int i=1;i<=m;++i) { if((ll)(A[i].x+0.5)>=1) { if(!bu[i]) { printf("NO\n"); return 0; } ans[i]=0; } } printf("YES\n"); int sum=0; for(int i=1;i<=m;++i) if(ans[i]) ++sum; printf("%d\n",sum); for(int i=1;i<=m;++i) if(ans[i]) printf("%d ",i); return 0; }