P5689 多叉堆

写在前面

• OI 生涯中 AC 的首道组合数学应用题。

• 开题 5min 发现规律，写了半下午代码，调了两天，然而甚至没过样例，心态崩了。几天之后重新写了一份代码才 AC。

• 虽然思维难度不大，但毕竟是联赛题，题目质量还是很高的。涉及到了很多组合数学的基础算法，写完之后感觉学到了很多。感觉这道题是道不错的组合数学入门题。接下来我将会尽量详细地进行讲解。

数据结构

只有根节点的答案有用。任何一个节点在更新完其父亲结点的值后，其本身的任何值将不会再有任何改动或贡献，因此用并查集维护即可，记得路径压缩。

算法思路

解决这道题的关键，在于连接两个节点时答案的更新和与答案相关的值的维护。

设 (a_i) 表示以 (i) 为根节点的子树的填数方案总数。

假设当前连接的两个节点的编号分别是 (u,v)（保证两个节点都是根节点），且本次操作需要从 (u) 接到 (v) 上去。

比较显而易见的一点是，更新答案时，节点 (v) 所处的位置，一定只能填 (0)。

维护 (w_i) 表示以 (i) 为根的树的重量，即节点数。将 (u) 连接到 (v) 上时，将 (w_v) 的值加上 (w_u)。

那么填入原来的 (u) 子树中的数字就有 (w_v - 1) 选 (w_u) 种不同的选择方案。剩下的数字填入原来的 (v) 子树中。又原本 (u,v) 子树中的填数总方案数分别为 (a_u,a_v)。那么根据乘法原理，就可以将 (a_v) 更新为 (a_v imes a_u imes inom{n}{m})

组合数求法

1. 杨辉三角，加法，可以取模。但是这是 P2822 的组合数求法，需要用到二维数组，数组大小开不下，只能过 50% 的数据。

2. 直接根据组合数计算公式：

[inom{n}{m} = frac{n!}{m!(n - m)!} ]

直接用计算式来求组合数。预处理出阶乘。等下，模意义下的乘法……要求逆元啊。线性递推求逆元可以参考 P3811 的题解。

这道题的除数是可以大到 (10^9 + 7) 的，因此不能直接递推预处理逆元，但是只会用到阶乘的逆元，预处理这个即可。

那么怎么线性递推呢？看这里：（为了更好体现逆元的形式，式子中保证了分数分子上都是 (1)）

[frac{1}{i!} = frac{1}{icdot(i - 1)!} = frac{1}{i}cdotfrac{1}{(i - 1)!} ]

[frac{1}{(i - 1)!} = icdotfrac{1}{i!} ]

因此，首先根据费马小定理推论用快速幂算出 (frac{1}{n!}) 随后倒序枚举预处理即可。

Tips

• 节点编号是从 (0) 开始的，记得初始化的时候把 (a_0,w_0) 也一并设为 (1)。（我卡了几天就是因为这个）
• 注意输入格式，别忘了强制在线。
• 处理阶乘的逆元记得要枚举到 (0)，不然 (m = 0) 的时候 (inom{n}{m}) 的值是错的。

Code

代码可读性还是很高的。

#include<bits/stdc++.h>
#define LL long long

using namespace std;

const int Maxn = 3e5 + 5;
const LL Mod = 1e9 + 7;

int n, q, opt, x, y;
int Ans;
LL a[Maxn] = {1};
int w[Maxn] = {1}; 

/*快速幂*/ 
inline LL qpow(LL b, LL p)
{
	if(p == 0) return 1;
	LL x = 1;
	for(; p;b *= b, b %= Mod,p >>= 1) if(p & 1) x *= b, x %= Mod;
	return x;
}

/*并查集*/
int fa[Maxn];

int find(int t)
{
	return fa[t] == t ? t : fa[t] = find(fa[t]);
}

/*数学*/
LL fac[Maxn] = {1};
LL invf[Maxn];

LL C(int N, int M)
{
	return 1ll * fac[N] % Mod * invf[M] % Mod * invf[N - M] % Mod;
}

/*预处理*/ 
void Setup()
{
	for(register int i = 1; i <= n; ++i)
	{
		fa[i] = i;
		w[i] = a[i] = 1;
		fac[i] = fac[i - 1] * i;
		fac[i] %= Mod;
	}
	invf[n] = qpow(fac[n], Mod - 2);
	for(register int i = n - 1; i >= 0; --i)
	{
		invf[i] = invf[i + 1] * (i + 1);
		invf[i] %= Mod;
	}
}

/*链接 处理答案*/
void line(int u, int v)
{
	w[v] += w[u];
	a[v] = a[v] * a[u] % Mod * C(w[v] - 1, w[u]) % Mod;
	fa[u] = v;
}

/*输出答案 更新强制在线值*/
void print(int t)
{
	Ans = (int)a[t];
	printf("%d
", Ans);
} 

/*快速读入*/ 
inline int read()
{
	int f = 1, w = 0; char ch = getchar();
	for(; (ch < '0') || (ch > '9'); ch = getchar()) if(ch == '-') f = -1;
	for(; (ch >= '0') && (ch <= '9'); ch = getchar()) w = (w << 3) + (w << 1) + (int)(ch ^ '0');
	return f * w;
}

int main()
{
	n = read(); q = read();
	Setup();
	while(q--)
	{
		opt = read();
		if(opt == 1)
		{
			x = (read() + Ans) % n;
			y = (read() + Ans) % n;
			x = find(x); y = find(y);
			line(x, y); 
		}
		else
		{
			x = (read() + Ans) % n;
			x = find(x);
			print(x);
		}
	}
	return 0;
}