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

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


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


#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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值