2021牛客暑期多校训练营9 C Cells

2021牛客暑期多校训练营9 C Cells

题意

给定n个点，分别在$(0,a_1),(0,a_2)...(0,a_n)$,现在这$n$个点只能向右或向下移动，要求这$n$个点分别走到$(1,0),(2,0)...(n,0)$且路径两两不相交的总方案数，注意$(0,a_i)$可以走到$(1,0),(2,0)...(n,0)$中的任何一个。

Sol

首先不相交路径数我们优先考虑$LGV$引理~~(当时不会，大雾)~~

根据$LGV$​,由于每个$(0,a_i)$​到$(1,0),(2,0)...(n,0)$​的路径数分别为$C_{a_i+1}^1,C_{a_i+2}^2,...,C_{a_i+n}^n$​

也即$\frac{(a_i+1)!}{1!a_i!},\frac{(a_i+2)!}{2!a_i!},...,\frac{(a_i+n)!}{n!a_i!}$

那么所对应的答案矩阵为
$$M=\left[\begin{array}{cccc}\frac{(a_1+1)!}{1!a_1!}& \frac{(a_1+2)!}{2!a_1!}& \cdots & \frac{(a_i+n)!}{n!a_n!}\\ \frac{(a_2+1)!}{1!a_2!}& \frac{(a_2+2)!}{2!a_2!}& \cdots & \frac{(a_2+n)!}{n!a_2!}\\ ⋮& ⋮& \ddots & ⋮\\ \frac{(a_n+1)!}{1!a_n!}& \frac{(a_n+2)!}{2!a_n!}& \cdots & \frac{(a_n+n)!}{n!a_n!}\end{array}\right]\text{\hspace{1em}(5)}$$

然后每一列提出个$\frac{1}{i!}$出来，再把上下约分一下
$$M=\prod _{i=1}^n\frac{1}{i!}\left[\begin{array}{cccc}(a_1+1)& (a_1+1)(a_1+2)& \cdots & (a_1+1)(a_1+2)...(a_1+n)\\ (a_2+1)& (a_2+1)(a_2+2)& \cdots & (a_2+1)(a_2+2)...(a_2+n)\\ ⋮& ⋮& \ddots & ⋮\\ (a_n+1)& (a_n+1)(a_n+2)& \cdots & (a_n+1)(a_n+2)...(a_n+n)\end{array}\right]\text{\hspace{1em}(5)}$$
以第一行为例

$(a_1+1)(a_1+2)-(a_1+1)=(a_1+1{)}^2$

$(a_1+1)(a_1+2)(a_1+3)-3(a_1+1)(a_1+2)+(a_1+1)=(a_1+1{)}^3$

以此类推:每一列都可以通过前几列华为$(a_1+1{)}^i$的形式，于是
$$M=\prod _{i=1}^n\frac{1}{i!}\left[\begin{array}{cccc}(a_1+1)& (a_1+1{)}^2& \cdots & (a_1+1{)}^n\\ (a_2+1)& (a_2+1{)}^2& \cdots & (a_2+1{)}^n\\ ⋮& ⋮& \ddots & ⋮\\ (a_n+1)& (a_n+1{)}^2& \cdots & (a_n+1{)}^n\end{array}\right]\text{\hspace{1em}(5)}$$
然后每行提一个公因式出来
$$M=\prod _{i=1}^n\frac{1}{i!}\prod _{i=1}^n(a_i+1)\left[\begin{array}{cccc}1& (a_1+1)& \cdots & (a_1+1{)}^{n-1}\\ 1& (a_2+1)& \cdots & (a_2+1{)}^{n-1}\\ ⋮& ⋮& \ddots & ⋮\\ 1& (a_n+1)& \cdots & (a_n+1{)}^{n-1}\end{array}\right]\text{\hspace{1em}(5)}$$
到这里就很清楚了，后面的行列式是一个范德蒙行列式
$$\begin{gathered} M=\prod _{i=1}^n\frac{1}{i!}\prod _{i=1}^n(a_i+1)\prod _{1\le i<j\le n}^n((a_j+1)-(a_i+1)) \\ =\prod _{i=1}^n\frac{1}{i!}\prod _{i=1}^n(a_i+1)\prod _{1\le i<j\le n}^n(a_j-a_i) \end{gathered}$$
于是用$O(n)$处理处${\prod }_{i=1}^n\frac{1}{i!}$和${\prod }_{i=1}^n(a_i+1)$

后面差的形式如果暴力则$O(n^2)$,显然不可，考虑用卷积$(FFT/NTT)$​ ，优化到$O(nlogn)$

为什么可以这样?一般的卷积是求系数相乘，怎么才能求减法呢?

由于卷积相乘幂次是相加的，于是可以把$a_i和-a_i$当做幂次，这样卷积完之后，遍历幂次的所有范围，如果系数不为0，说明这个幂次存在也即$(a_j-a_i)$存在。

下面再考虑一个问题，卷积求的是任意两个的乘积，而我们要求的是有顺序的$1\le i<j\le n$。

首先由于存在负数，于是我们把负数加上一个偏移量($a_i$的最大值记为$M$)，于是我们幂次有$a_i$和
$M-a_i$ ,而我们最后卷积完会得到的所有幂次范围为$[0,2\ast M]$ ,这是注意题目有个条件$1\le i<j\le n,a_i<a_j$​,于是这样卷完$(a_j-a_i)$等价于$a_j+M-a_i>M$ 于是所有符合的幂次我们就从$M+1$到$2\ast M$枚举，其对应的系数就为这个差值出现的次数，快速幂求一下就可以了。

#include <bits/stdc++.h>
#define ll long long
#define lll __int128
using namespace std;
const  int P = 998244353,G=3; // P=1004535809 ord(P)=3
const int pi=acos(-1);
inline ll read()
{
     ll x = 0, f = 1;
     char ch = getchar();
    while(ch < '0' || ch > '9') {if(ch == '-')f = -1;ch = getchar();}
    while(ch >= '0' && ch <= '9') {x = x * 10 + ch - '0';ch = getchar();}
    return x * f;
}

const int N=1e6+10,M=1e6;
ll n,m;
ll rev[1<<22],bit,tot;
ll a[1<<22],b[1<<22];
ll qmi(ll x,ll y)
{
    ll res=1;
    while(y)
    {
        if(y&1) res=res*x%P;
        x=x*x%P;
        y>>=1;
    }
    return res;
}
ll inv(ll x)
{
    return qmi(x,P-2);
}
void NTT(ll a[],int type)
{
    for(int i=0;i<tot;i++)
    if(i<rev[i]) swap(a[i],a[rev[i]]);

    for(int mid=1;mid<tot;mid<<=1)
    {
        ll wn=qmi(G,(P-1)/(mid*2));
        if(type==-1) wn=qmi(wn,P-2);
        for(int i=0,len=mid<<1;i<tot;i+=len)
        {
            ll w=1;
            for(int j=0;j<mid;j++,w=w*wn%P)
            {
                ll x=a[i+j],y=w*a[i+j+mid]%P;
                a[i+j]=(x+y)%P,a[i+j+mid]=(x-y+P)%P;
            }
        }
    }
    if(type==-1)
    {
        ll inv_=inv(tot);
        for(int i=0;i<tot;i++)
        a[i]=a[i]*inv_%P;
    }
}
int main()
{
     n=read();
    ll ans=1;
    for(int i=1;i<=n;i++)
    {
        ll x;
        x=read();
        ans=ans*(x+1)%P;
        a[x]=1,b[M-x]=1;
    }
    ll k1=1,k2=1;
    for(int i=1;i<=n;i++)
    {
        k1=k1*i%P;
        k2=k2*k1%P;
    }
    ans=ans*inv(k2)%P; 
    //cout<<ans<<endl;

    tot=1<<21,bit=21;
    for(int i=0;i<tot;i++) 
    rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
    NTT(a,1),NTT(b,1);
    for(int i=0;i<tot;i++) a[i]=a[i]*b[i]%P;
    NTT(a,-1);
      
    for(int i=1e6+1;i<=M*2;i++)
    {
        if(a[i])
        {  
            ans=ans*qmi(i-1e6,a[i])%P; //求的是 (a_j - a_i ) (0<i<j<=n&&a_i<a_j) 
                                       //所以枚举>M的数就是(a_j-a_i)
        }
    }
    cout<<ans<<endl;
    puts("");
    
    return 0;

    
}

​