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

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

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

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

#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]<<" ";
}
此文章已被阅读次数:正在加载...更新于