[斯特林数][NTT][HDU4372][WC2019雅礼集训]Count the buildings

HDU4372

考虑最高的点,他左边有x-1个极高点,右边有y-1个极高点
一个极高点和下一个极高点(不包含下一个)这一部分可以看做一个圆排列(不是排列因为极高点的位置固定),那么就是要求 n − 1 n-1 n1个数分成 x + y − 2 x+y-2 x+y2个圆排列的方案数,即 s n − 1 , x + y − 2 s_{n-1,x+y-2} sn1,x+y2
然后考虑这些圆排列要选x-1个放到最高点左边,这里只用选出来就行了,选出来之后的排列顺序是固定的
a n s = s n − 1 , x + y − 2 ∗ ( x − 1 x + y − 2 ) ans=s_{n-1,x+y-2}*(_{x-1}^{x+y-2}) ans=sn1,x+y2(x1x+y2)
预处理stirling数可以直接 n 2 n^2 n2预处理,也可以用倍增NTT O ( n l o g n ) O(nlogn) O(nlogn)预处理

Code(倍增NTT):

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(int *f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f+1,f+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}
	inline void mull(int *a,int *b,int n){
		init(n);
		ntt(a,n,1);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);
	}
}
using namespace Ntt;
int fac[N],ifac[N];
int a[N],b[N],c[N];
inline void init_fac(int n){
	fac[0]=ifac[0]=1;
	for(int i=1;i<=n;i++) fac[i]=mul(fac[i-1],i);
	ifac[n]=ksm(fac[n],mod-2);
	for(int i=n-1;i;i--) ifac[i]=mul(ifac[i+1],i+1);
}
inline void solve(int n){
	if(n==1){a[1]=1;return;}
	if(n&1){
		solve(n-1);
		for(int i=n;i;i--) a[i]=add(a[i-1],mul(a[i],n-1));
		return;
	}
	else{
		solve(n>>1);
		int lim=1;
		while(lim<=n) lim<<=1;
		int mid=n>>1;
		for(int i=0,p=1;i<=mid;i++,Mul(p,mid)) c[i]=mul(a[i],fac[i]),b[i]=mul(p,ifac[i]);
		for(int i=mid+1;i<lim;i++) c[i]=b[i]=0;
		reverse(b,b+mid+1);
		mull(b,c,lim);
		for(int i=0;i<=mid;i++) b[i]=mul(b[i+mid],ifac[i]);
		for(int i=mid+1;i<lim;i++) b[i]=0;
		mull(a,b,lim);
	}
}
inline ll C(int n,int m){if(n<m) return 0;return mul(fac[n],mul(ifac[m],ifac[n-m]));}
int main(){
	int n=read(),A=read(),B=read();
	init_w();init_fac(n);
	if(n==1){cout<<(A==1&&B==1);return 0;}
	solve(n-1);
	cout<<mul(a[A+B-2],C(A+B-2,A-1));
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值