[BZOJ4835] [Lydsy1704月赛] 遗忘之树 [乱搞]

原创于 2019-04-20 16:32:51 发布 · 274 阅读

本内容遵循CC 4.0 BY-SA版权协议

本文详细解析了BZOJ-4835题目的解法，利用图论和动态规划技术，在O(nlogn)的时间复杂度下，解决了关于图的重心变化问题。通过判断子树大小，确定连边方式，避免重心改变，实现了高效的计数算法。

Link
BZOJ - https://www.lydsy.com/JudgeOnline/problem.php?id=4835

直接在图 G 上 dp 。
考虑图 G 上的 u → v
u 能够连几条边到 v 呢？重点在于连边之后重心会不会变
如果图 G 中 v 为根的子树大小卜大于等于以 u 为根的子树的一半那么咋连边都没有问题
否则这个子树大小一定等于 $u$ 为根子树大小的一半，此时你只能向以 v 为根的子树里面编号大于 u 的点连边。
（否则重心就会变嗷）
以此为依据计数即可，复杂度是 $O(n\log n)$ 的。
某个子树有两个重心的情况需要特别分类讨论。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cctype>
using namespace std;
#define R register
const int MAXN = 1e5 + 5;
const int MOD = 1e9 + 7;
int T, n, m, tot, Root;
int head[MAXN], nxt[MAXN], to[MAXN], Deg[MAXN], Size[MAXN];
long long F[MAXN];
#define add_edge(a, b) nxt[++tot] = head[a], head[a] = tot, to[tot] = b
void Sizel(const int& u)
{
	Size[u] = 1;
	for (int v, i = head[u]; i; i = nxt[i])
	{
		v = to[i];
		Sizel(v);
		Size[u] += Size[v];
	}
}
int Calc(const int& v, const int& u)
{
	int Ret = (v > u);
	for (int i = head[v]; i; i = nxt[i])
	{
		Ret += Calc(to[i], u);
	}
	return Ret;
}
void Solve(const int& u)
{
	for (int Half = (Size[u] >> 1), v, i = head[u]; i; i = nxt[i])
	{
		v = to[i];
		Solve(v);
		if ((!(Size[u]&1)) && (Size[v] == Half))
		{
			F[u] = F[u] * F[v] % MOD * Calc(v, u) % MOD;
		}
		else
		{
			F[u] = F[u] * F[v] % MOD * Size[v] % MOD;
		}
	}
}
int main()
{
	scanf("%d", &T);
	while (T--)
	{
		tot = 0;
		scanf("%d%d", &n, &m);
		fill(head, head + n + 1, 0); 
		fill(Deg, Deg + n + 1, 0);
		fill(F, F + n + 1, 1);
		for (R int u, v, i = 1; i <= m; ++i)
		{
			scanf("%d%d", &u, &v);
			add_edge(u, v);
			++Deg[v];
		}
		for (R int i = 1; i <= n; ++i)
		{
			if (!Deg[i])
			{
				Root = i;
				break;
			}
		}
		Sizel(Root);
		Solve(Root);
		printf("%lld\n", F[Root]);
	}
	return 0;
}