Codeforces 960G：Bandit Blues（倍增FFT/第一类斯特林数）

传送门

题解：
答案等于 [n−1a+b−2]∗(a+b−2a−1) [ n − 1 a + b − 2 ] ∗ ( a + b − 2 a − 1 )

将最多的袋子剔除后，每个能看到的袋子与他挡住的袋子形成圆排列。我们从这 a+b−2 a + b − 2 个圆排列中取出 a−1 a − 1 个放到左边。

注意 [nm]=[xm]xn⎯⎯⎯ [ n m ] = [ x m ] x n ¯ 。我们直接求 xn⎯⎯⎯ x n ¯ 即可。

O(nlog2n) O ( n log 2 ⁡ n ) 可以启发式合并多项式。

可以 O(nlogn) O ( n log ⁡ n ) ，记 l=n2 l = n 2 。我们先求 xl⎯⎯ x l ¯ ，得到：

F=∑i=0laixi F = ∑ i = 0 l a i x i

要求

G=∑i=0nai(x+l)i G = ∑ i = 0 n a i ( x + l ) i

二项式展开：

G=∑i=0nai(x+l)i=∑i=0nai∑j=0i(ij)xjli−j=∑j=0nxj∑i=jn(ij)aili−j G = ∑ i = 0 n a i ( x + l ) i = ∑ i = 0 n a i ∑ j = 0 i ( i j ) x j l i − j = ∑ j = 0 n x j ∑ i = j n ( i j ) a i l i − j

记 bi=[xi]Gj!,Ai=aii!,Bi=ljj! b i = [ x i ] G j ! , A i = a i i ! , B i = l j j ! ，那么：

bj=∑i=jnAiBi−j b j = ∑ i = j n A i B i − j

我们把 B B reverse，发现这就是个卷积形式，直接迭代下去做即可，时间复杂度：
T(n)=T(n2)+O(nlogn)=O(nlogn) T ( n ) = T ( n 2 ) + O ( n log ⁡ n ) = O ( n log ⁡ n )

#include 
using namespace std;
typedef long long LL;
const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}
inline void W(int x) {
    static int buf[50];
    if(!x) {putchar('0'); return;}
    if(x<0) {putchar('-'); x=-x;}
    while(x) {buf[++buf[0]]=x%10; x/=10;}
    while(buf[0]) {putchar(buf[buf[0]--]+'0');}
}

const int N=8e5+50, mod=998244353, G=3;
inline int add(int x,int y) {return (x+y>=mod) ? (x+y-mod) : (x+y);}
inline int dec(int x,int y) {return (x-y<0) ? (x-y+mod) : (x-y);}
inline int mul(int x,int y) {return (LL)x*y%mod;}
inline int power(int a,int b,int rs=1) {for(;b;b>>=1,a=mul(a,a)) if(b&1) rs=mul(rs,a); return rs;}

int n,A,B,k,pos[N],w[N],a[N],b[N],tp[N],fac[N],ifac[N],pw[N];
inline int C(int ai,int bi) {
    if(aireturn 0; 
    return mul(fac[ai],mul(ifac[bi],ifac[ai-bi]));
}
inline void init(int len) {
    for(k=1;k<=len;k<<=1);
    for(int i=1;i1) ? ((pos[i>>1]>>1)^(k>>1)) : (pos[i>>1]>>1);
}
inline void dft(int *ai) {
    for(int i=1;iif(pos[i]>i) swap(ai[pos[i]],ai[i]);
    for(int bl=1;bl1) {
        int tl=bl<<1, wn=power(G,(mod-1)/tl);
        w[0]=1; for(int i=1;i1],wn);
        for(int bg=0;bgfor(int j=0;jint &t1=ai[bg+j],&t2=ai[bg+j+bl],t=mul(t2,w[j]);
                t2=dec(t1,t); t1=add(t1,t);
            }
    }
}
inline void mul(int *ai,int *bi) {
    dft(ai); dft(bi);
    for(int i=0;i1,bi+k);
    const int inv=power(k,mod-2);
    for(int i=0;iinline void solve(int len ) {
    if(len==1) {a[1]=1; return;}
    if(len&1) {
        solve(len-1);
        for(int i=len;i>=1;i--) a[i]=add(a[i-1],mul(a[i],len-1)); 
    } else {
        solve(len>>1); int l2=len/2; pw[0]=1; 
        init(len);
        for(int i=1;i<=l2;i++) pw[i]=mul(pw[i-1],l2);
        for(int i=0;i<=l2;i++) tp[i]=mul(a[i],fac[i]);
        for(int i=0;i<=l2;i++) b[i]=mul(pw[i],ifac[i]);
        for(int i=l2+1;i0;
        reverse(b,b+l2+1);
        mul(tp,b); 
        for(int i=0;i<=l2;i++) b[i]=b[i+l2], b[i]=mul(b[i],ifac[i]);
        for(int i=l2+1;i0;
        mul(b,a);
    }
}
int main() {
    n=rd(), A=rd(), B=rd();
    if(n==1) {cout<<(A==1 && B==1); return 0;}
    fac[0]=1;
    for(int i=1;i<=n;i++) fac[i]=mul(fac[i-1],i);
    ifac[n]=power(fac[n],mod-2);
    for(int i=n-1;~i;i--) ifac[i]=mul(ifac[i+1],i+1);
    solve(n-1); 
    cout<2],C(A+B-2,A-1));
}