hdu 5829 Rikka with Subset (2016多校第八场1009) FFT

公式非常好求，我们只需要求每个点第k大的时候的贡献，排好序后第二个元素的贡献zuhe（i-1,k-1）*a[i]*2^(n-i),我们观察可知(i-1)!*a[i]*2^(n-i)是不变的，1/k!每次都一样也很容易所以就剩下1/(i-k)!，我们用b[i]/(i-k)!表示我们注意到把i-k倒过来变成(n-i+k)然后（n-i+k+i）=n+k所以这可以用FFT处理，注意要取模。

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std ;
typedef long long LL ;
typedef long double db;
#define clr( a , x ) memset ( a , x , sizeof a )
typedef long long LL;
const int maxn=410005;
const int g=3;
const int mod=998244353;
int G[30],nG[30];
void fft(int *x,int n,int fl=1){
    for (int i=(n>>1),j=1;j<n;j++){
        if (i<j) swap(x[i],x[j]);
        int k;
        for (k=(n>>1);i&k;i^=k,k>>=1);
        i^=k;
    }
    int now=0;
    for (int m=2;m<=n;m<<=1){
        int w; now++; if (fl==1) w=G[now]; else w=nG[now];
        for (int i=0;i<n;i+=m){
            int cur=1;
            for (int j=i;j<i+(m>>1);j++){
                int u=x[j],v=1ll*x[j+(m>>1)]*cur%mod;
                x[j]=(u+v)%mod; x[j+(m>>1)]=(u-v+mod)%mod;
                cur=1ll*cur*w%mod;
            }
        }
    }
}
int x[maxn],y[maxn],z[maxn];
int mm(int n,int m){
    LL s=1;
    LL k=n;
    while(m){
        if(m&1){
            s=s*k;
            s%=mod;
        }
        k=k*k;
        k%=mod;
        m>>=1;
    }
    return s;
}
int jie[maxn],jie1[maxn],fa[maxn];
void init(){
    jie[0]=1;
    jie1[0]=1;
    fa[0]=1;
    for(int i=1;i<maxn;i++){
        jie[i]=((LL)jie[i-1]*i)%mod;
        jie1[i]=mm(jie[i],mod-2);
        fa[i]=fa[i-1]*2;
        fa[i]%=mod;
    }
    int now=(mod-1)/2,ng=mm(g,mod-2),len=0;
    while (now%2==0){
        len++; G[len]=mm(g,now); nG[len]=mm(ng,now);  now>>=1;
    }
}
int a[maxn];
int ans[maxn];
int cmp(int x,int y){
    return x>y;
}
int main()
{
    init();
    int t,n;
    cin>>t;
    while (t--) {
        scanf("%d",&n);
        ans[1]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        sort(a+1,a+1+n,cmp);
        for(int i=1;i<=n;i++){
            a[i]=((LL)a[i]*jie[i-1])%mod;
            a[i]=((LL)a[i]*fa[n-i])%mod;
        }
        int p=1;
        while(p<=2*n) p=p*2;
        x[0]=0;
        for(int i=1;i<=n;i++) x[i]=a[i];
        for(int i=n+1;i<p;i++){
            x[i]=y[i]=0;
        }
        for(int j=0;j<=n;j++) y[j]=jie1[n-j];
        fft(x,p);
        fft(y,p);
        for(int i=0;i<p;i++){
            x[i]=((LL)x[i]*y[i])%mod;
        }
        fft(x,p,-1);
        int w=mm(p,mod-2);
        for(int i=1;i<=n;i++){
            ans[i]=((LL)x[n+i]*jie1[i-1])%mod;
            ans[i]=((LL)ans[i]*w)%mod;
            ans[i]+=ans[i-1];
            ans[i]%=mod;
            ans[i]=(ans[i]+mod)%mod;
            printf("%d ",ans[i]);
        }
        printf("\n");
    }
}