二项式反演与斯特林反演

二项式反演

如果 $f(m)={\sum }_{i=m}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)g(i)$
根据二项式反演：
$$g(i)=\sum _{k=i}^n(-1{)}^{k-i}(\genfrac{}{}{0ex}{}{k}{i})f(k)$$
另一种形式罢了。
正常版本：
$$\begin{gathered} f(n)=\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)g(i) \\ g(n)=\sum _{i=0}^n(-1{)}^{n-i}(\genfrac{}{}{0ex}{}{n}{i})f(i) \end{gathered}$$
$$\begin{gathered} f(n)=\sum _{i=0}^n(-1{)}^i(\genfrac{}{}{0ex}{}{n}{i})g(i) \\ g(n)=\sum _{i=0}^n(-1{)}^i\left(\genfrac{}{}{0ex}{}{n}{i}\right)f(i) \end{gathered}$$

就是在容斥。例如：

1

n 个人枪手，每名枪手随机瞄准其余的 n-1 个人中的一人，有 p 的概率命中。问同时打一枪后，恰有 i 个人死的概率。

记 $f(S)$ 表示恰为 S 死的概率，$g(S)$ 表示死的是 S 子集的概率。那么显然 $g$ 可以快速求出（每个人都向某个集合瞄准），并且 $g(S)={\sum }_{T\subseteq S}f(T)$，那么容斥容易得到 $f(S)={\sum }_{T\subseteq S}(-1{)}^{|S|-|T|}g(T)$。

发现对于 |S| 相同的集合，$f(S)$ 也一样。于是用 |S| 代替 S，上述两个式子就变成了 $g(n)={\sum }_{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)f(i)$，$f(n)={\sum }_{i=0}^n(-1{)}^{n-i}(\genfrac{}{}{0ex}{}{n}{i})g(i)$，和二项式定理是一样的。

斯特林数

• 第一类斯特林数：把n个标号元素排成m个环的方案数。
  $$S_1(n,m)=S_1(n-1,m-1)+(n-1)\cdot S_1(n-1,m)$$有一种组合意义是 $n$ 个元素的排列，有 $m$ 个前缀最大值的方案数。可以自然推出：${\sum }_{i=0}^n{S}_1(n,i)=n!$
  根据递推式还可以写出 $S_1(n)$ 的生成函数：$S_1(n)={\prod }_{i=0}^{n-1}(x+i)$。我们用分治 FFT 在 nlog²n 的时间计算某一行。

• 第二类斯特林数：把n个标号元素划分为m个相同集合的方案数。
  $$S_2(n,m)=S_2(n-1,m-1)+m\cdot S_2(n-1,m)$$
  容斥计算：
  $$\begin{gathered} S_2(n,m)=\frac{1}{m!}\sum _{i=0}^m(-1{)}^i(\genfrac{}{}{0ex}{}{m}{i})(m-i{)}^n \\ =\sum _{i=0}^m\frac{(-1{)}^i}{i!}\cdot \frac{(m-i{)}^n}{(m-i)!} \end{gathered}$$可以O(n)算某一项，nlogn 计算某一行。

小技巧

$$n^k=\sum _{i=0}^n{S}_2(k,i)\left(\genfrac{}{}{0ex}{}{n}{i}\right)i!$$
左边相当于把k个标号球放进n个标号盒子里。右边枚举放在那些盒子里，由于盒子有标号所以乘上阶乘，再乘上斯特林数。

注意到上式的 i 枚举到 n，实际上可以枚举到 k，因为后面的组合数乘阶乘可以看做一个 i 次下降幂，如果 k>n 那么这个下降幂会变成 0，如果 k<n，那么 i 在 大于 k 的时候前面的斯特林数就会变成 0。也就是：
$$n^k=\sum _{i=0}^n{S}_2(k,i)\cdot n^{i\downarrow }$$
我们把这个式子变得和底数没什么关系了。有什么好处呢？如果我们要维护 $\sum a_i^k$，k 是常数，我们只需要维护 $\sum a_i^{k\downarrow }$，最后再 O(k) 计算答案。注意到 $(x+1{)}^{k\downarrow }=x^{k\downarrow }+k\cdot x^{(k-1)\downarrow }$（考虑组合意义），因此 $\sum a_i^{k\downarrow }$ 这个式子是可以在 O(k) 的时间变成 $\sum (a_i+1{)}^{k\downarrow }$。

2

给 n, k 和一个长度为 n 的数字串。要求对每一个 i, 输出 ${\sum }_{j=1}^{i}F(j,i)$。其中 F(l, r) 为 l 到 r 内所有数字之和的 k 次方。
$n\le 50000,k\le 100$

直接二项式定理复杂度是 O(nk^2)，用上面的式子复杂度是 O(nk*10) 的。

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define fir first
#define sec second
#define ld long double
using namespace std;
const int N=100010,mod=1e9+7;
typedef pair <int,int> P;
char s[N];
ll S[110][110],f[110];
int a[N],n,k;
int read()
{
	int x=0;char c=getchar(),flag='+';
	while(!isdigit(c)) flag=c,c=getchar();
	while(isdigit(c)) x=x*10+c-'0',c=getchar();
	return flag=='-'?-x:x;
}
void add()
{
	for(int i=k;i>0;i--) f[i]=(f[i]+f[i-1]*i)%mod;
}
int main()
{
	int T=read();
	for(int i=1;i<=100;i++)
	{
		S[i][1]=1;
		for(int j=2;j<=i;j++) S[i][j]=(S[i-1][j-1]+S[i-1][j]*j)%mod;
	}
	while(T--) //f[i]=sigma{F(j,x)的i次下降幂} 
	{
		memset(f,0,sizeof(f));
		n=read(),k=read();
		scanf("%s",s+1);
		for(int i=1;i<=n;i++) a[i]=s[i]-'0';
		for(int i=1;i<=n;i++)
		{
			f[0]=(f[0]+1)%mod;
			for(int j=1;j<=a[i];j++) add();
			int ans=0;
			for(int j=1;j<=k;j++) ans=(ans+S[k][j]*f[j])%mod;
			cout<<ans<<' ';
		}
	}
	return 0;
}
/*by DT_Kang*/

斯特林反演：

$$\begin{gathered} f[n]=\sum _{i=1}^n{S}_2(n,i)\cdot g[i] \\ g[n]=\sum _{i=1}^n(-1{)}^{n-i}\cdot S_1(n,i)\cdot f[i] \end{gathered}$$
$$\begin{gathered} f[m]=\sum _{i=m}^n{S}_2(i,m)\cdot g[i] \\ g[m]=\sum _{i=m}^n(-1{)}^{i-m}{S}_1(i,m)\cdot f[i] \end{gathered}$$
相比二项式反演就是把组合数换成斯特林数。

3

定义两个结点数相同的图 G1 与图 G2 的异或为一个新的图 G, 其中如果 (u, v) 在 G1 与G2 中的出现次数之和为 1, 那么边 (u, v) 在 G 中, 否则这条边不在 G 中. 现在给定 s 个结点数相同的图 G1…Gs, 设 S = {G1, G2, … , Gs}, 请问 S 有多少个子集的异或为一个连通图?

10的贝尔数是可以接受的。
用总方案减去不合法的，我们要算的就是g[i]表示有i个联通块的方案数。枚举了点集划分以后，不同的点集之间的边是不能存在的，相同点集的边随便选，记为f。但是这样没有保证一个点集的点联通。因此实际上：$$f[m]=\sum _{i=m}^n{S}_2(i,m)\cdot g[i]$$
斯特林反演一下：
$$g[m]=\sum _{i=m}^n(-1{)}^{i-m}{S}_1(i,m)\cdot f[i]$$
只要求g[1]，即
$$\begin{gathered} g[1]=\sum _{i=1}^n(-1{)}^{i-1}{S}_1(i,1)\cdot f[i] \\ =\sum _{i=1}^n(-1{)}^{i-1}(i-1)!\cdot f[i] \end{gathered}$$

那么问题就变成了如何求f。我们把边压成一个longlong，枚举了集合划分之后，相当于有一些数，求有多少个子集异或位0。根据线性基的理论，我们求出一个线性基，枚举线性基外面的任意一个集合，都能在线性基里找到恰好一个元素，它们的异或为0。

4

雅礼集训 方阵

给定一个方阵，每个位置可以填 [1,c] 中的任意一个数，要求填完之后不能有两行或两列完全相同，求方案数。$n,m\le 5000$

这种行列都有限制的题目往往是先考虑一边，也就是让行都不等价，这样列可能存在等价的情况。设 f_n 表示 n 列互补等价的方案数，g_n 表示任意的方案数，那么实际上 $g_n={\sum }_{i=1}^n{f}_i{S}_2(n,i)$，反演一下可以得到 $f_n={\sum }_{i=1}^n(-1{)}^{n-i}{g}_i{S}_1(n,i)$。

容易发现，第二类斯特林数总是和等价类划分之类的问题相关。

5

计算
$$\sum _{i=0}^n\sum _{j=0}^{n}S(i,j)\cdot j!\cdot 2^j$$

解法1：
第二类斯特林数可以展开：
$$\begin{gathered} S(n,m)=\frac{1}{m!}\sum _{i=0}^m(-1{)}^i\cdot \frac{m!}{i!(m-i)!}\cdot (m-i{)}^n \\ =\sum _{i=0}^m\frac{(-1{)}^i}{i!}\cdot \frac{(m-i{)}^n}{(m-i)!} \end{gathered}$$
是一个卷积。因此我们可以在 $O(n\log n)$ 的时间里计算 $S(n,0),S(n,1),\dots ,S(n,n)$。

代回原式：
$$\begin{gathered} \sum _{i=0}^n\sum _{j=0}^{n}S(i,j)\cdot j!\cdot 2^j=\sum _{i=0}^n\sum _{j=0}^{n}j!\cdot 2^j\sum _{k=0}^j\frac{(-1{)}^k}{k!}\cdot \frac{(j-k{)}^i}{(j-k)!} \\ =\sum _{j=0}^{n}j!\cdot 2^j\sum _{k=0}^j\frac{(-1{)}^k}{k!}\cdot \frac{{\sum }_{i=0}^n(j-k{)}^i}{(j-k)!} \end{gathered}$$

解法2：
考虑第二类斯特林数的意义，是把 $n$ 个不同的球放入 $m$ 个相同盒子里的方案数。由于盒子非空，因此乘上一个 $j!$ 以后就是放进不同盒子的方案数了。
我们令 $g_n$ 表示 ${\sum }_{i=0}^{n}S(n,i)\cdot i!\cdot 2^i$。如果不考虑 $2^i$，也就是把 $n$ 个不同的球放进任意多个不同的盒子里的方案数。我们枚举最后一个盒子放几个球，可以写出递推式：
$$\begin{gathered} g_n=\sum _{i=1}^{n}2\cdot \left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot g_{n-i} \\ \frac{g_n!}{n!}=\sum _{i=1}^n\frac{2}{i!}\cdot \frac{g_{n-i}}{(n-i)!} \end{gathered}$$
设：
$$\begin{gathered} f(x)=\sum _{i=0}^{\infty }\frac{g_i}{i!}{x}_i \\ h(x)=\sum _{i=1}^{\infty }\frac{2}{i!}{x}_i \end{gathered}$$
则：
$$\begin{gathered} f(x)=f(x)\ast h(x)+1 \\ f(x)=\frac{1}{1-h(x)} \end{gathered}$$
为什么要加一？因为 $f(x)$ 和 $g(x)$ 卷出来是没有常数项的。
然后就是多项式求逆啦。