其他分享
首页 > 其他分享> > Codeforces960G Bandit Blues

Codeforces960G Bandit Blues

作者:互联网

Problem

Codeforces

Solution

先找到序列中 \(n\) 的位置,那么在 \(n\) 之前必须有 \(a-1\) 个前缀最大值,之后有 \(b-1\) 个后缀最大值。

设 \(f[i][j]\) 表示长度为 \(i\) 的排列,有 \(j\) 个前缀最大值的方案数。

那么\(ans=\sum_{i=1}^n f[i-1][a-1]\times f[n-i][b-1]\times \binom n {i-1}\)

枚举最小值的位置,那么当且仅当它在第一个位置上时才会贡献一个前缀最大值,则 \(f[i][j]=f[i-1][j-1]+(i-1)f[i-1][j]\)。不难发现这其实是第一类斯特林数的递推式。

怎么理解呢?不妨记第 \(i\) 个前缀最大值的出现位置为 \(p_i\),把\([p_i,p_{i+1})\)视为一个盒子,盒子内的排列的方案数就是圆排列的方案数,所以这就是第一类斯特林数。

前后其实是一个对称的问题,那把它们放在一起考虑,答案就是 \(f[n-1][a+b-2]\binom {a+b-2} {b-1}\),即把这些盒子排在一起,然后再选 \(b-1\) 个盒子放到后面去。

第一类斯特林数可以用它的生成函数来做,倍增FFT加速即可做到\(O(n\log n)\)。

Code

#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=300010,mod=998244353,G=3;
template <typename Tp> int getmin(Tp &x,Tp y){return y<x?x=y,1:0;}
template <typename Tp> int getmax(Tp &x,Tp y){return y>x?x=y,1:0;}
template <typename Tp> void read(Tp &x)
{
    x=0;char ch=getchar();int f=0;
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') ch=getchar(),f=1;
    while(ch>='0'&&ch<='9') x=x*10+(ch-'0'),ch=getchar();
    if(f) x=-x;
}
int n,a,b,N,l,ans,fac[maxn],inv[maxn],rev[maxn],f[maxn],g[maxn];
int pw[maxn],ta[maxn],tb[maxn];
int pls(int x,int y){return x+y>=mod?x+y-mod:x+y;}
int dec(int x,int y){return x-y<0?x-y+mod:x-y;}
int c(int n,int m){return m>n?0:(ll)fac[n]*inv[m]%mod*inv[n-m]%mod;}
int power(int x,int y)
{
    int res=1;
    for(;y;y>>=1,x=(ll)x*x%mod)
      if(y&1)
        res=(ll)res*x%mod;
    return res;
}
void NTT(int *a,int f)
{
    for(int i=1;i<N;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
    for(int i=1;i<N;i<<=1)
    {
        int gn=power(G,(mod-1)/(i<<1));
        for(int j=0;j<N;j+=(i<<1))
        {
            int g=1,x,y;
            for(int k=0;k<i;++k,g=(ll)g*gn%mod)
            {
                x=a[j+k];y=(ll)g*a[j+k+i]%mod;
                a[j+k]=pls(x,y);a[j+k+i]=dec(x,y);
            }
        }
    }
    if(f==-1)
    {
        int inv=power(N,mod-2);reverse(a+1,a+N);
        for(int i=0;i<N;i++) a[i]=(ll)a[i]*inv%mod;
    }
}
void solve(int n)//x^n
{
    if(n==1){f[1]=1;return ;}
    int h=n>>1;solve(n>>1);
    for(N=1,l=0;N<=n;N<<=1) ++l;
    for(int i=1;i<N;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=1;i<=h;i++) pw[i]=(ll)pw[i-1]*h%mod;
    for(int i=0;i<=h;i++) ta[i]=(ll)fac[i]*f[i]%mod;
    for(int i=0;i<=h;i++) tb[i]=(ll)pw[i]*inv[i]%mod;
    for(int i=h+1;i<N;i++) ta[i]=tb[i]=0;
    reverse(tb,tb+h+1);
    NTT(ta,1);NTT(tb,1);NTT(f,1);
    for(int i=0;i<N;i++) g[i]=(ll)ta[i]*tb[i]%mod;
    NTT(g,-1);
    for(int i=0;i<=h;i++) g[i]=(ll)g[i+h]*inv[i]%mod;
    for(int i=h+1;i<N;i++) g[i]=0;
    NTT(g,1);
    for(int i=0;i<N;i++) f[i]=(ll)f[i]*g[i]%mod;
    NTT(f,-1);
    if(n&1)
    {
        for(int i=n;i;i--) f[i]=pls(f[i-1],(ll)f[i]*(n-1)%mod);
        f[0]=(ll)f[0]*(n-1)%mod;
    }
}
int main()
{
    read(n);read(a);read(b);fac[0]=pw[0]=1;
    if(a<1||b<1){puts("0");return 0;}
    if(n==1){printf("%d\n",(a==1&&b==1));return 0;}
    for(int i=1;i<=n;i++) fac[i]=(ll)fac[i-1]*i%mod;
    inv[n]=power(fac[n],mod-2);
    for(int i=n-1;~i;i--) inv[i]=(ll)inv[i+1]*(i+1)%mod;
    solve(n-1);
    ans=(ll)f[a+b-2]*c(a+b-2,a-1)%mod;
    printf("%d\n",ans);
    return 0;
}

标签:ch,int,最大值,Tp,Bandit,Blues,Codeforces960G,ll,mod
来源: https://www.cnblogs.com/totorato/p/10359105.html