The 2020 ICPC Asia Macau Regional Contest A. Accelerator（分治+NTT）

最新推荐文章于 2022-10-27 17:58:00 发布

秋天俯身采种子

最新推荐文章于 2022-10-27 17:58:00 发布

阅读量994

点赞数 1

分类专栏： ACM补题文章标签：分治 NTT

本文链接：https://blog.csdn.net/Fighting_Peter/article/details/120879300

版权

ACM补题专栏收录该内容

66 篇文章 1 订阅

订阅专栏

本文介绍了一种高效计算多项式乘法的方法，利用快速幂和线性基理论，实现了在大规模整数模意义下计算指定指数系数的高效算法。通过实例展示了如何初始化、计算和操作线性基，并应用于解决实际问题，如计算特定指数的乘积之和。

摘要由CSDN通过智能技术生成

A. Accelerator

实质上就是求n个多项式相乘
$(a_1+x)(a_2+x)\dots(a_n+x)$
对于 $x^m]$ 的系数即是从 $n$ 个选出 $n - m$ 的乘积之和。

每次从 $n$ 个选出 $n - m$ 的概率显然为
$\frac{\binom{n}{m}}{n!}$

然后期望累加即可。

或者说求
$(1+a_1x)(1+a_2x)\dots(1+a_nx)$
对于 $x^m]$ 的系数即是从 $n$ 个选出 $m$ 的乘积之和。

#include<bits/stdc++.h>

using namespace std;
using ll=long long;

const int N=300010;
const int P=998244353;
const int G=3,Gi=332748118;
const int mod=998244353;

int qmi(int a,int b)
{
    int v=1;
    while(b)
    {
        if(b&1) v=1ll*v*a%P;
        a=1ll*a*a%P;
        b>>=1;
    }
    return v;
}
int rev[N];
void NTT(int *a,int n,int inv)
{
    for(int i=0;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
    
    for(int mid=1;mid<n;mid<<=1)
    {
        int Wn=qmi((inv==1)?G:Gi,(P-1)/(mid<<1));
        for(int i=0;i<n;i+=(mid<<1))
        {
            int w=1;
            for(int j=0;j<mid;j++,w=1ll*w*Wn%P)
            {
                int x=a[i+j],y=1ll*w*a[i+j+mid]%P;
                a[i+j]=(x+y)%P;
                a[i+j+mid]=(x-y+P)%P;
            }
        }
        
    }
    if(inv==-1) 
    {
        int invn=qmi(n,P-2);
        for(int i=0;i<n;i++) a[i]=1ll*a[i]*invn%P;
    }
}
//=======================================================
int fact[N],infact[N];
void init()
{
    fact[0]=1;
    for(int i=1;i<=100000;i++) fact[i]=1ll*i*fact[i-1]%mod;
    infact[100000]=qmi(fact[100000],mod-2);
    
    for(int i=99999;i;i--) infact[i]=1ll*infact[i+1]*(i+1)%mod;
}
int n,A[N],B[N],C[N];
int Mul(int *a,int *b,int n,int m,int *ans)
{
    int bit=0,num=1;
    while(num<n+m+1) num<<=1,bit++;
    
    for(int i=0;i<num;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
    for(int i=0;i<=n;i++) A[i]=a[i];
    for(int i=0;i<=m;i++) B[i]=b[i];
    for(int i=n+1;i<num;i++) A[i]=0;
    for(int i=m+1;i<num;i++) B[i]=0;
    NTT(A,num,1);NTT(B,num,1);
    for(int i=0;i<num;i++) C[i]=1ll*A[i]*B[i]%mod;
    NTT(C,num,-1);
    for(int i=0;i<=n+m;i++) ans[i]=C[i];
    return n+m;
}

int pool[N<<1],tot;
struct Node
{
    int *p,n;
    void init(int x)
    {
        this->n=1;
        p=pool+tot;
        for(int i=0;i<=n;i++) p[i]=0;
        p[0]=x;p[1]=1;
        tot+=n+1;
    }
    void mul(const Node&o)
    {
        n=Mul(p,o.p,n,o.n,p);
    }
};
Node solve(int l,int r)
{
    Node ans;
    if(l==r)
    {
        int x;
        cin>>x;
        ans.init(x);
        return ans;
    }
    int mid=l+r>>1;
    ans=solve(l,mid);
    ans.mul(solve(mid+1,r));
    return ans;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    init();
    int Tc;
    cin>>Tc;
    while(Tc--)
    {
        cin>>n;
        Node res=solve(1,n);
        
        ll ans=0;
        for(int i=1;i<=n;i++)
            ans=(ans+1ll*res.p[n-i]*fact[i]%mod*fact[n-i]%mod)%mod;
        ans=ans*infact[n]%mod;
        printf("%lld\n",ans);
    }
    return 0;
    
}