CF286E

发表于 更新于 分类于 本文字数： 498 阅读时长 ≈ 2 分钟

题意是要找出一个最小集合，使得从其中选取任意次数的元素之和等于或者大于。

容易发现一个可能解就是题目给定的集合，如果这不是一个解，那么答案就是无解。

接下来考虑的是如何缩减这个集合，显然如果有，那么就可以从解集中删除，而且容易发现不可能有某个元素仅由三个及以上的元素相加才可得，因为，那么必有。

所以我们只要一次NTT，直接卷自己即可。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include <bits/stdc++.h>
using namespace std;

using ll=long long;

const int mod=998244353;
const int g=3,gi=332748118,NR=1<<21;

int n,m,a[(int)1e6+9],A[NR],B[NR],rev[NR],lim=1,bit,ans;
int powmod(int a,int b){
	int res=1;
	for(;b>0;b>>=1,a=1ll*a*a%mod)if(b&1)res=1ll*res*a%mod;
	return res;
}

void NTT(int *a,int n,int type){
	for(int i=0;i<n;++i)if(i<rev[i])swap(a[i],a[rev[i]]);
	for(int i=1;i<n;i<<=1){
		int gn=powmod(type?g:gi,(mod-1)/(i<<1));
		for(int j=0;j<n;j+=(i<<1)){
			ll g0=1;
			for(int k=0;k<i;++k,g0=1ll*g0*gn%mod){
				int x=a[j+k],y=g0*a[i+j+k]%mod;
				a[j+k]=(x+y)%mod,a[i+j+k]=(x-y+mod)%mod;
			}
		}
	}
	if(type==1)return ;
	int invn=powmod(n,mod-2);
	for(int i=0;i<n;++i)a[i]=1ll*a[i]*invn%mod;
}

bitset<(int)1e6+9> vis;


int main(){
	cin.tie(nullptr)->sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;++i)cin>>a[i];
	ans=n;
	while(lim<=2*m)lim<<=1,bit+=1;
	for(int i=0;i<lim;++i)rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
    for(int i=1;i<=n;++i)A[a[i]]=B[a[i]]=1;
    NTT(A,lim,1),NTT(B,lim,1);
	for(int i=0;i<lim;++i)A[i]=1ll*A[i]*B[i]%mod;
	NTT(A,lim,0);
	for(int i=1;i<=n;++i)if(A[a[i]])ans-=1,A[a[i]]=0,vis[a[i]]=1;
	for(int i=0;i<=m;++i)if(A[i]){
		cout<<"NO\n";
		return 0;
	}
	cout<<"YES\n"<<ans<<"\n";
	for(int i=1;i<=n;++i)if(!vis[a[i]])cout<<a[i]<<" ";
}