考虑转化题意,我们发现其实就是找一个长度为(n)的全排列,使得这个排列有(A)个前缀最大值,(B)个后缀最大值,求方案数
我们考虑把最大值拎出来单独考虑,同时定义一些数的顺序排列为单调块(随便取的名字)
考虑在这个最大值左边有(A-1)个单调块,右边有(B-1)个单调块,如果这些块在左右两边按序排好的话就是一种合法方案
那我们只需要找出(A+B-2)个单调块,并且将其中拿出(A-1)个放在左边,因此答案有一项就是(C_{A+B-2}^{A-1})
考虑怎么从除了最大值外的(n-1)个数里挑出单调块,我们仔细分析一下,发现单调块的个数其实就是圆排列的个数
因为对于任意一个圆排列(a_1,a_2,cdots,a_k),我们必然可以拿出其中的最大值(a_{mx}),然后(a_{mx},a_{mx+1},a_k,cdots,a_1,a_{mx-1})就是一个单调块
不难发现这种关系唯一对应,所以答案已经出来了,就是([_{A+B-2}^{n-1}]cdot C_{A+B-2}^{A-1})
那么对于Luogu P4609 [FJOI2016]建筑师 我们(O(n^2))预处理出第一类斯特林数和组合数的值,然后(O(1))计算就好了
CODE
#include<cstdio>
#define RI register int
#define CI const int&
using namespace std;
const int N=50000,M=200,mod=1e9+7;
int t,n,a,b,C[M+5][M+5],s[N+5][M+5];
inline int sum(CI a,CI b)
{
int t=a+b; return t>=mod?t-mod:t;
}
inline int min(CI a,CI b)
{
return a<b?a:b;
}
inline void init(void)
{
RI i,j; for (s[0][0]=i=1;i<=N;++i) for (j=min(i,M);j;--j)
s[i][j]=sum(s[i-1][j-1],1LL*(i-1)*s[i-1][j]%mod);
for (i=0;i<=M;++i) for (C[i][0]=C[i][i]=j=1;j<i;++j)
C[i][j]=sum(C[i-1][j],C[i-1][j-1]);
}
int main()
{
for (init(),scanf("%d",&t);t;--t)
{
scanf("%d%d%d",&n,&a,&b); if (!a||!b||n-1<a+b-2) { puts("0"); continue; }
printf("%d
",1LL*s[n-1][a+b-2]*C[a+b-2][a-1]%mod);
}
return 0;
}
考虑对于CF960G Bandit Blues 显然我们不能暴力求出所有的斯特林数,而是只求出一个即可
但是对于一个我们也没有太好的公式去算,因此我们类似于第二类斯特林数,可以用生成函数来算出一行的值
通过数学推导或者OIES,我们得到了第一类斯特林数的生成函数:
[S_n=prod_{i=0}^{n-1}(x+i)
]
最后(x^k)的系数就是([^n_k]),用分治NTT做到(O(nlog^2 n))。有(nlog n)的做法但懒得写了(分治NTT实在好写)
CODE
#include<cstdio>
#include<vector>
#include<algorithm>
#define RI register int
#define CI const int&
#define pb push_back
using namespace std;
typedef vector <int> VT;
const int N=100005,mod=998244353;
int n,a,b;
inline int fact(CI n,int ret=1)
{
for (RI i=1;i<=n;++i) ret=1LL*ret*i%mod; return ret;
}
inline int quick_pow(int x,int p=mod-2,int mul=1)
{
for (;p;p>>=1,x=1LL*x*x%mod) if (p&1) mul=1LL*mul*x%mod; return mul;
}
inline int C(CI n,CI m)
{
return 1LL*fact(n)*quick_pow(1LL*fact(m)*fact(n-m)%mod)%mod;
}
class Poly_Solver
{
private:
int rev[N<<2],lim,p;
inline int sum(CI a,CI b)
{
int t=a+b; return t>=mod?t-mod:t;
}
inline int sub(CI a,CI b)
{
int t=a-b; return t<0?t+mod:t;
}
inline void swap(int& x,int& y)
{
int t=x; x=y; y=t;
}
inline void init(CI n)
{
for (lim=1,p=0;lim<=n;lim<<=1,++p);
for (RI i=0;i<lim;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<p-1);
}
inline void NTT(int *f,CI opt)
{
RI i; for (i=0;i<lim;++i) if (i<rev[i]) swap(f[i],f[rev[i]]);
for (i=1;i<lim;i<<=1)
{
int m=i<<1,D=quick_pow(3,~opt?(mod-1)/m:mod-1-(mod-1)/m);
for (RI j=0;j<lim;j+=m)
{
int W=1; for (RI k=0;k<i;++k,W=1LL*W*D%mod)
{
int x=f[j+k],y=1LL*f[i+j+k]*W%mod;
f[j+k]=sum(x,y); f[i+j+k]=sub(x,y);
}
}
}
if (!~opt)
{
int Inv=quick_pow(lim); for (RI i=0;i<lim;++i) f[i]=1LL*f[i]*Inv%mod;
}
}
inline VT merge(const VT& VA,const VT& VB)
{
static int A[N<<2],B[N<<2],n,m; VT VC; RI i;
for (n=VA.size(),i=0;i<n;++i) A[i]=VA[i];
for (m=VB.size(),i=0;i<m;++i) B[i]=VB[i];
init(n+m); fill(A+n,A+lim,0); fill(B+m,B+lim,0);
for (NTT(A,1),NTT(B,1),i=0;i<lim;++i) A[i]=1LL*A[i]*B[i]%mod;
for (NTT(A,-1),i=0;i<n+m-1;++i) VC.pb(A[i]); return VC;
}
public:
inline VT solve(CI l,CI r)
{
VT p; if (l==r) return p.pb(l),p.pb(1),p;
int mid=l+r>>1; return merge(solve(l,mid),solve(mid+1,r));
}
}P;
int main()
{
scanf("%d%d%d",&n,&a,&b); if (!a||!b||n-1<a+b-2) return puts("0"),0;
if (n==1) return puts("1"),0; VT ans=P.solve(0,n-2);
return printf("%d",1LL*ans[a+b-2]*C(a+b-2,a-1)%mod),0;
}