[BZOJ4835] [Lydsy1704月賽] 遺忘之樹 [亂搞]

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


直接在圖 G 上 dp 。
考慮圖 G 上的 u → v
u 能夠連幾條邊到 v 呢 ?重點在於連邊之後重心會不會變
如果圖 G 中 v 爲根的子樹大小卜大於等於以 u 爲根的子樹的一半那麼咋連邊都沒有問題
否則這個子樹大小一定等於 uu 爲根子樹大小的一半,此時你只能向以 v 爲根的子樹裏面編號大於 u 的點連邊。
(否則重心就會變嗷)
以此爲依據計數即可,複雜度是 O(nlogn)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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章