标签:signed begin next sig 整数 http c++ highlight swa
fft+快速幂,需要不少优化才能过
#include<bits/stdc++.h> #define ll long long #define rep(ii,a,b) for(int ii=a;ii<=b;++ii) #define per(ii,a,b) for(int ii=b;ii>=a;--ii) #define forn(i,x,g,e) for(int i=g[x];i;i=e[i].next) #define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0) #define ull unsigned long long #define fi first #define se second #define mp make_pair #define pii pair<ll,ll> #define all(x) x.begin(),x.end() #define show(x) cout<<#x<<"="<<x<<endl #define showa(a,b) cout<<#a<<‘[‘<<b<<"]="<<a[b]<<endl #define show2(x,y) cout<<#x<<"="<<x<<" "<<#y<<"="<<y<<endl #define show3(x,y,z) cout<<#x<<"="<<x<<" "<<#y<<"="<<y<<" "<<#z<<"="<<z<<endl #define show4(w,x,y,z) cout<<#w<<"="<<w<<" "<<#x<<"="<<x<<" "<<#y<<"="<<y<<" "<<#z<<"="<<z<<endl #define show5(v,w,x,y,z) cout<<#v<<"="<<v<<" "<<#w<<"="<<w<<" "<<#x<<"="<<x<<" "<<#y<<"="<<y<<" "<<#z<<"="<<z<<endl #define showa2(x,a,b) cout<<#x<<": ";rep(i,a,b) cout<<x[i]<<‘ ‘;cout<<endl using namespace std; int n,k; const double pi=acos(-1.0); struct cp{double x,y;}; inline cp operator*(cp a,cp b){return {a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};} inline cp operator+(cp a,cp b){return {a.x+b.x,a.y+b.y};} inline cp operator-(cp a,cp b){return {a.x-b.x,a.y-b.y};} const int maxl=1e6+10; int la,lb; bool a[maxl<<2]; bool ret[maxl<<2]; class fourier{public: int rev[maxl<<2],len,pw; void init(int n){ len=1,pw=0; while(len<=n) len<<=1,pw++; rep(i,0,len-1) rev[i]=0; rep(i,0,len-1) rev[i]=(rev[i>>1]>>1)|((i&1)<<(pw-1)); } cp c1[maxl<<2],c2[maxl<<2]; cp wt[maxl<<2]; void init_0(int ml=maxl){ for(int mid=1;mid<2*maxl;mid<<=1){ // wt.resize(mid*2+1); cp wn1={cos(pi/mid),sin(pi/mid)}; wt[mid]=(cp){1,0}; rep(j,1,mid-1) wt[mid+j]=wt[mid+j-1]*wn1; } } void transform(cp*a,int flag){ rep(i,0,len-1) if(i<rev[i]) swap(a[i],a[rev[i]]); for(int mid=1;mid<len;mid<<=1){ for(int r=mid<<1,j=0;j<len;j+=r){ for(int k=0;k<mid;++k){ cp wn=wt[mid+k]; if(flag==-1) wn.y=-wn.y; cp y=wn*a[mid+j+k]; a[j+k+mid]=a[j+k]-y,a[j+k]=a[j+k]+y; } } } if(flag==-1) rep(i,0,len-1) a[i].x/=len; } void fix1(){while(la&&!ret[la-1])--la;} void fix2(){while(lb&&!a[lb-1])--lb;} //@传入整数,进行卷积@ void mul(bool *a,bool *b){ init(la+lb); rep(i,0,la-1)c1[i].x=a[i],c1[i].y=0; rep(i,la,len-1)c1[i].x=c1[i].y=0; rep(i,0,lb-1)c2[i].x=b[i],c2[i].y=0; rep(i,lb,len-1)c2[i].x=c2[i].y=0; transform(c1,1);transform(c2,1); rep(i,0,len-1) c1[i]=c1[i]*c2[i]; transform(c1,-1); la=len; rep(i,0,len-1) a[i]=c1[i].x>=0.5; fix1(); } void sqr(bool *a){ init(2*lb); rep(i,0,lb-1)c1[i].x=a[i],c1[i].y=0; rep(i,lb,len-1)c1[i].x=c1[i].y=0; transform(c1,1); rep(i,0,len-1) c1[i]=c1[i]*c1[i]; transform(c1,-1); lb=len; rep(i,0,len-1) a[i]=c1[i].x>=0.5; fix2(); } void pow(int k){ fix1();la=1,ret[0]=1; while(k){ if(k&1) mul(ret,a); sqr(a); k>>=1; } } }fft; int s[2000]; int main(){ ios::sync_with_stdio(false);cin.tie(0); cin>>n>>k; for(int i=1;i<=n;i++){ cin>>s[i]; a[s[i]]=1; } sort(s+1,s+1+n); fft.init_0(s[n]*k+1); lb=1001; fft.pow(k); int len=min(la,s[n]*k+1); for(int i=k;i<len;i++) if(ret[i]) cout<<i<<" "; }
标签:signed begin next sig 整数 http c++ highlight swa
原文地址:https://www.cnblogs.com/nervendnig/p/11589611.html