[NowCoder牛客]2021NOIP提高组模拟赛第二场T3——树数树（启发式合并堆）

description

【题目描述】
牛牛有一棵 n 个点的有根树，根为 1。
我们称一个长度为 m 的序列 a 是好的，当且仅当：
• $\forall i\in (1,m]$， $a_i$为 $a_{i-1}$的祖先或 $a_{i-1}$是 $a_i$ 的祖先
• $\forall 1\le i<j\le m,a_i\ne a_j$
你需要帮助牛牛求出最长的序列长度

【输入格式】
第一行一个正整数 T，表示数据组数。
对于每组数据第一行一个正整数 n。
接下来 n - 1行，每行两个正整数 u,v，表示树上的一条边

【输出格式】
𝐓 行，每行一个整数表示每组数据的答案
【样例输入】

1 
8
5 3
1 5
4 5
2 5
1 6
8 7
7 6

【样例输出】

7

【数据范围】

对于 100% 的数据， 1 ≤ T ≤ 5， 2 ≤ n ≤ 10⁵， 1 ≤ u, v ≤ n， u ≠ v，输入保证是一棵树

PS : 祖先不仅仅是父亲哦

solution

考场上把祖先当成了父亲，那不就是树的直径吗？？我直接来

好的一遍WA，然后开始怀疑数据——最后好的，打扰了

手玩发现，一个点可以连接其子树的两条链

略微想了一下$n\le 3000$的$d{p}_{i,j}$以$i$为根的子树用了$i$及以上的祖先共$j$个，然后最大值$dp$，但是就是没写出来。。。

回归正题，没错一个节点确实可以将子树中的两个序列拼接起来

且处理完父亲节点后就没必要管儿子节点了

所以用堆来维护，点和点的子树所能组成的序列，从下往上合并，每次相当于是子树全合并给父亲，然后父亲选最长的两个子序列进行拼接合并

启发式合并即可

code

#include <queue>
#include <cstdio>
#include <vector>
using namespace std;
#define maxn 100005
vector < int > G[maxn];
priority_queue < int > q[maxn];
int id[maxn];

int merge( int x, int y ) {
	if( q[x].size() < q[y].size() ) swap( x, y );
	while( ! q[y].empty() ) q[x].push( q[y].top() ), q[y].pop();
	return x;
}

void dfs( int u, int fa ) {
	id[u] = u;
	while( ! q[id[u]].empty() ) q[id[u]].pop();
	for( auto v : G[u] )
		if( v ^ fa ) dfs( v, u ), id[u] = merge( id[u], id[v] );
	if( q[id[u]].empty() ) q[id[u]].push( 1 );
	else {
		int w = q[id[u]].top(); q[id[u]].pop();
		if( ! q[id[u]].empty() ) w += q[id[u]].top(), q[id[u]].pop();
		q[id[u]].push( w + 1 );
	}
}

int main() {
	int T, n, u, v;
	scanf( "%d", &T );
	while( T -- ) {
		scanf( "%d", &n );
		for( int i = 1;i <= n;i ++ ) G[i].clear();
		for( int i = 1;i < n;i ++ ) {
			scanf( "%d %d", &u, &v );
			G[u].push_back( v );
			G[v].push_back( u );
		}
		dfs( 1, 0 );
		printf( "%d\n", q[id[1]].top() );
	}
	return 0;
}