HDU 5829 快速数论变换

题目链接

---

题意：给你$n$个数和一个$k$，对于这$n$ 个数组成的集合的值，定义它的值为所有子集的和，而一个子集的值是该子集前$k$大元素值的和（若$k>|S|$，则为$S$集合中所有元素值的和），问对于$k=1$ ~ $n$，输出每个$k$值对应的集合的值。

思路：
如果我们知道了每一个子集的第$k$大元素的和，那么对于任意的$k$，所需要求的集合的值就是$1$~$k$的前缀和。

设该集合每一个子集的第$k$大元素加起来的和为$f(n)$

很明显，我们需要先将元素从大到小排序。

然后我们考虑每一个元素的贡献。
易得：
$$f(n)=\sum _{i=k}^{n}A[i]\ast C_{i-1}^{k-1}\ast 2^{n-i}$$

解释：当第$i$个元素为某个集合的第$k$大元素时，该集合一定有$(k-1)$个比$A[i]$大的元素，而$n$个元素中一共有$i-1$个比$A[i]$大的元素，故需要乘上一个组合数。
而另外还有$n-i$个比$A[i]$小的元素，此时这些元素可取可不取，每个元素有两种选择，故乘上一个排列。

试着将组合数打开，得：
$$f(n)=\frac{1}{(k-1)!}\ast \sum _{i=k}^{n}A[i]\ast 2^{n-i}\ast \frac{(i-1)!}{(i-k)!}$$

观察易发现和式的前两项满足卷积形式，故尝试换元，使$\sum$的$i$从0开始，同时式子里每一个存在$i$的地方变成$i+k$，得：

$$f(n)=\frac{1}{2^k(k-1)!}\sum _{i=0}^{n-k}A[i+k]\ast 2^{n-i}\ast \frac{(i+k-1)!}{i!}$$

令
$$x[i]=\frac{2^{n-i}}{i!}$$
$$y[i]=A[i]\ast (i-1)!$$

则：
$$f(n)=\frac{1}{2^k(k-1)!}\sum _{i=0}^{n-k}x[i]\ast y[i+k]$$

此时要想办法将$x$和$y$的下标和凑成$n-k$
考虑翻转$y$数组，再将$y$整体左移一位，得：

$$f(n)=\frac{1}{2^k(k-1)!}\sum _{i=0}^{n-k}x[i]\ast y[n-k-i]$$

然后用$NTT$或者$FFT$维护卷积即可。

---

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;

const ll mod = 998244353;
const ll N   = 262144;
const ll G   = 3;
const ll K   = 17;
ll a[N<<1],x[N<<1],y[N<<1],fac[N<<1],inv[N<<1],Inv[N<<1],inv_2;

ll fast_pow(ll n,ll m){
    ll res = 1;
    while(m){
        if(m&1) res = res*n%mod;
        n = n*n%mod;
        m >>= 1;
    }
    return res;
}

void rader(ll y[], int len){
    for(int i = 1, j = len / 2; i < len - 1; i++){
        if(i < j) swap(y[i], y[j]);
        int k = len / 2;
        while(j >= k) {
            j -= k;
            k /= 2;
        }
        if(j < k) j += k;
    }
}

void ntt(ll y[], int len, int on){
    rader(y, len);
    for(int h = 2; h <= len; h <<= 1){
        ll wn = fast_pow(G,(mod-1)/h);
        if(on == -1) wn = fast_pow(wn, mod-2);
        for(int j = 0; j < len; j += h) {
            ll w = 1;
            for(int k = j; k < j + h / 2; k++) {
                ll u = y[k];
                ll t = w * y[k + h / 2] % mod;
                y[k] = (u + t) % mod;
                y[k+h/2] = (u - t + mod) % mod;
                w = w * wn % mod;
            }
        }
    }
    if(on == -1) {
        ll t = fast_pow(len, mod - 2);
        for(int i = 0; i < len; i++)
            y[i] = y[i] * t % mod;
    }
}

void init(){
    inv_2 = fast_pow(2,mod-2);
    fac[0] = 1;
    for(ll i=1 ;i<N ;i++)    fac[i] = 1LL*i*fac[i-1]%mod;
    inv[N-1] = fast_pow(fac[N-1],mod-2);
    for(ll i=N-2 ;i>=0 ;i--) inv[i] = inv[i+1]*1LL*(i+1)%mod;
}

bool cmp(ll& p,ll& q){return p>q;}

void construct_Fun(ll &n){
    scanf("%I64d",&n);
    memset(x,0,sizeof(x));memset(y,0,sizeof(y));
    for(ll i=1 ;i<=n;i++) scanf("%I64d",&a[i]);
    sort(a+1,a+1+n,cmp);
    for(ll i=0 ;i<n ;i++) x[i] = 1LL*fast_pow(2,n-i)*inv[i]%mod;
    for(ll i=1 ;i<=n;i++) y[i] = 1LL*a[i]*fac[i-1]%mod;
    for(ll i=1 ;i<=n/2;i++) swap(y[i],y[n-i+1]);
    for(ll i=0 ;i<n ;i++) y[i] = y[i+1];
    y[n] = 0;
}

void solve(ll n){
    ll m = 1;
    while(m<=(n<<1)) m<<=1;
    ntt(x,m,1);ntt(y,m,1);
    for(ll i=0 ;i<m ;i++){
        x[i] = x[i]*y[i]%mod;
    }
    ntt(x,m,-1);

    ll pre = 0;ll r = inv_2;
    for(ll k=1 ;k<=n ;k++){
        ll res = r*inv[k-1]%mod*x[n-k]%mod;
        pre = (pre+res)%mod;
        printf("%I64d ",pre);
        r = r*inv_2%mod;
    }
    puts("");
}

int main(){
    init();
    ll T;scanf("%I64d",&T);
    while(T--){
        ll n;
        construct_Fun(n);
        solve(n);
    }
    return 0;
}