luogu P4609 [FJOI2016]建筑师 & CF960G Bandit Blues

背景：

没有啊。

题目传送门：

https://www.luogu.org/problem/P4609
https://www.luogu.org/problem/CF960G

$\text{luogu P4609}$题意：

$n$个建筑，每一个建筑的高度都是互不相同的数$[1,n]$且互不相同。问有多少种排列方式使得从左往右能看到$A$个建筑，从右往左能看到$B$个建筑。

$\text{luogu P4609}$思路：

考虑高度为$n$的一定会被两边同时看到，因此它是一个分界线。
考虑最后的形状一定形如$A+B-1$块，且每一块的开头都会挡住后面这一块的建筑但不会挡住下一块的开头，而高度为$n$的独成一块，可以不用考虑它。因此问题就变成了将$n-1$个数分成$A+B-2$块，相当于将$n-1$个数分成$A+B-2$个圆环的方案数，显然这是一个第一类斯特林数，故方案数为$S_{n-1,A+B-2}$。而我们考虑从$A+B-2$块中选出$A-1$块放在高度为$n$的建筑前面，方案数就为$C_{A+B-2,A-1}$。
因此
$$ans=S_{n-1,A+B-2}{C}_{A+B-2,A-1}$$
预处理第一类斯特林数和组合数即可。

$\text{luogu P4609}$代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define mod 1000000007
using namespace std;
	int s[50010][210],c[210][210];
	int a,b,n;
void init()
{
	s[0][0]=1;
	for(int i=1;i<=50000;i++)
	{
		if(i<=200) s[i][i]=1;
		for(int j=1;j<min(i,201);j++)
			s[i][j]=((LL)s[i-1][j-1]+(LL)(i-1)*s[i-1][j]%mod)%mod;
	}
	c[0][0]=1;
	for(int i=1;i<=200;i++)
	{
		c[i][0]=c[i][i]=1;
		for(int j=1;j<i;j++)
			c[i][j]=((LL)c[i-1][j-1]+c[i-1][j])%mod;
	}
}
int main()
{
	int T;
	init();
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d %d %d",&n,&a,&b);
		if(a+b-1>n) printf("0\n"); else printf("%d\n",(LL)s[n-1][a+b-2]*c[a+b-2][a-1]%mod);
	}
}

$\text{CF960G}$题意：

其实本质是一样的。
当然数据范围变大了。

$\text{CF960G}$思路：

用分治$\text{NTT}$处理第一类斯特林数即可。
注意分治$\text{NTT}$时记录一下层数，记录在外面，以此不用在递归里面开数组，浪费栈空间。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
const LL mod=998244353,G=3,inv_G=332748118;
using namespace std;
	int r[400010],f[400010],a[20][400010],b[20][400010];
	int n,A,B,l,limit;
int dg(int x,int k)
{
	if(!k) return 1;
	int op=dg(x,k>>1);
	if(k&1) return (LL)op*op%mod*x%mod; else return (LL)op*op%mod;
}
int get_inv(int x)
{
	return dg(x,mod-2);
}
void init(int n)
{
	limit=1,l=0;
	while(limit<(n<<1))
		limit<<=1,l++;
	for(int i=1;i<limit;i++)
		r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void NTT(int *now,int limit,int op)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		int wn=dg(op==1?G:inv_G,(mod-1)/(mid<<1));
		for(int j=0;j<limit;j+=(mid<<1))
		{
			int w=1;
			for(int k=0;k<mid;k++,w=((LL)w*wn)%mod)
			{
				int x=now[j+k],y=(LL)w*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
		}
	}
	if(op==1) return;
	int INV=get_inv(limit);
	for(int i=0;i<limit;i++)
		now[i]=(LL)now[i]*INV%mod;
}
void poly_CDQ(int *f,int dep,int l,int r)
{
	if(l==r)
	{
		f[0]=l;f[1]=1;
		return;
	}
	int mid=(l+r)>>1;
	init(r-l+1+1);
	for(int i=0;i<limit;i++)
		a[dep][i]=b[dep][i]=0;
	poly_CDQ(a[dep],dep+1,l,mid),poly_CDQ(b[dep],dep+1,mid+1,r);
	init(r-l+1+1);
	NTT(a[dep],limit,1),NTT(b[dep],limit,1);
	for(int i=0;i<limit;i++)
		f[i]=(LL)a[dep][i]*b[dep][i]%mod;
	NTT(f,limit,-1);
}
int C(int n,int m)
{
	int tot1=1,tot2=1;
	for(int i=m+1;i<=n;i++)
		tot1=(LL)tot1*i%mod;
	for(int i=1;i<=n-m;i++)
		tot2=(LL)tot2*i%mod;
	return (LL)tot1*get_inv(tot2)%mod;
}
int main()
{
	scanf("%d %d %d",&n,&A,&B);
	if(!A||!B||A+B-1>n)
	{
		printf("0");
		return 0;
	}
	if(n==1)
	{
		printf("1");
		return 0;
	}
	poly_CDQ(f,0,0,n-1-1);
	printf("%d",(LL)f[A+B-2]*C(A+B-2,A-1)%mod);
}