CF960G

首先我们考虑$n$的情况,显然以$n$为分界线可以将整个序列分成两部分,就像这样:

那么我们考虑:在这个东西前面才会有前缀最大的统计,在这个东西后面才会有后缀最大的统计

这样就剩下了$n-1$个元素,而我们需要把这$n-1$个元素分成$A+B-2$个集合,然后把每个集合的最大的一个放在一端,然后剩余部分全排列

设一个集合中有$k$个元素,那么剩余元素全排列的方案数就是$(k-1)!$

考虑这也是对$k$个元素放在一个圆排列上的方案数,因此我们能搞出一个第一类斯特林数的表达式:

$S_{1}(n-1,A+B-2)$

然后考虑这几个集合在左右的分布,有:$C_{A+B-2}^{A-1}$

因此最终答案即为$S_{1}(n-1,A+B-2)C_{A+B-2}^{A-1}$

然后第一类斯特林数可以倍增$O(nlog_{2}n)$求解,因此能通过这题:

// luogu-judger-enable-o2
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define ll unsigned long long
#define uint unsigned int 
using namespace std;
const uint mode=998244353;
uint pow_mul(uint x,uint y)
{
    uint ret=1;
    while(y)
    {
        if(y&1)ret=1ll*ret*x%mode;
        x=1ll*x*x%mode,y>>=1;
    }
    return ret;
}
uint inv[300005],minv[300005],mul[300005];
void init()
{
    inv[0]=inv[1]=minv[0]=minv[1]=mul[0]=mul[1]=1;
    for(uint i=2;i<=300000;i++)
    {
        inv[i]=1ll*(mode-mode/i)*inv[mode%i]%mode;
        minv[i]=1ll*minv[i-1]*inv[i]%mode;
        mul[i]=1ll*mul[i-1]*i%mode;
    }
}
uint MOD(uint x,uint y)
{
    return x+y>=mode?x+y-mode:x+y;
}
uint to[(1<<20)+5];
uint n,a,b;
void NTT(uint *a,uint len,int k)
{
    for(uint i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);
    for(uint i=1;i<len;i<<=1)
    {
        uint w0=pow_mul(3,(mode-1)/(i<<1));
        for(uint j=0;j<len;j+=(i<<1))
        {
            uint w=1;
            for(uint o=0;o<i;o++,w=1ll*w*w0%mode)
            {
                ll w1=a[j+o],w2=1ll*a[j+o+i]*w%mode;
                a[j+o]=MOD(w1,w2)%mode,a[j+o+i]=MOD(mode-w2,w1);
            }
        }
    }
    if(k==-1)
    {
        uint inv=pow_mul(len,mode-2);
        for(uint i=1;i<(len>>1);i++)swap(a[i],a[len-i]);
        for(uint i=0;i<len;i++)a[i]=1ll*a[i]*inv%mode;
    }
}
uint A[(1<<20)+5],B[(1<<20)+5],C[(1<<20)+5];
uint tempt[(1<<20)+5],tempf[(1<<20)+5],tempg[(1<<20)+5],tempG[(1<<20)+5],tempF[(1<<20)+5];
uint ret[(1<<20)+5];
void mult(uint *f,uint *g,uint len)
{
    uint lim=1,l=0;
    while(lim<=2*(len+1))lim<<=1,l++;
    for(uint i=0;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(l-1)));
    for(uint i=0;i<lim;i++)A[i]=B[i]=0;
    for(uint i=0;i<=len;i++)A[i]=f[i],B[i]=g[i];
    for(uint i=0;i<lim;i++)C[i]=0;
    NTT(A,lim,1),NTT(B,lim,1);
    for(uint i=0;i<lim;i++)C[i]=1ll*A[i]*B[i]%mode;
    NTT(C,lim,-1);
}
void solve(uint *g,uint dep)
{
    if(!dep){g[0]=1;return;}
    if(dep==1)
    {
        g[0]=0,g[1]=1;
        return;
    }
    if(dep&1)
    {
        solve(g,dep-1);
        for(uint i=1;i<=dep;i++)tempG[i]=C[i-1];
        for(uint i=0;i<dep;i++)tempG[i]=MOD(tempG[i],1ll*C[i]*(dep-1)%mode);
        for(uint i=0;i<=dep;i++)g[i]=tempG[i];
    }else
    {
        uint mid=(dep>>1);
        solve(g,mid);
           uint temp=1;
        for(uint i=0;i<=mid;i++,temp=1ll*temp*mid%mode)tempg[i]=1ll*g[mid-i]*mul[mid-i]%mode,tempf[i]=1ll*temp*minv[i]%mode;
        mult(tempg,tempf,mid);
        for(uint i=0;i<=mid;i++)tempF[i]=1ll*C[mid-i]*minv[i]%mode;
        mult(tempF,g,mid);
        for(uint i=0;i<=dep;i++)g[i]=C[i];
    }
}
uint get_C(uint x,uint y)
{
    if(x<y)return 0;
    return 1ll*mul[x]*minv[y]%mode*minv[x-y]%mode;
}
int main()
{
    scanf("%u%u%u",&n,&a,&b);init();
    solve(ret,n-1);
    printf("%u\n",1ll*ret[a+b-2]*get_C(a+b-2,a-1)%mode);
    return 0;
}

 

posted @ 2019-07-08 18:46  lleozhang  Views(179)  Comments(0Edit  收藏  举报
levels of contents