树数树
- description
- solution
- code
description
【题目描述】
牛牛有一棵 n 个点的有根树,根为 1。
我们称一个长度为 m 的序列 a 是好的,当且仅当:
• ∀𝑖∈(1,𝑚]∀𝑖∈(1, 𝑚]∀i∈(1,m], aia_iai为 ai−1a_{i-1}ai−1的祖先或 ai−1a_{i-1}ai−1是 aia_iai 的祖先
• ∀1≤𝑖<𝑗≤𝑚,𝑎𝑖≠𝑎𝑗∀1 ≤ 𝑖 < 𝑗 ≤ 𝑚, 𝑎_𝑖 ≠ 𝑎_𝑗∀1≤i<j≤m,ai=aj
你需要帮助牛牛求出最长的序列长度
【输入格式】
第一行一个正整数 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 ≤ 105, 1 ≤ u, v ≤ n, u ≠ v,输入保证是一棵树
PS : 祖先不仅仅是父亲哦
solution
考场上把祖先当成了父亲,那不就是树的直径吗??我直接来
好的一遍WA,然后开始怀疑数据——最后好的,打扰了
手玩发现,一个点可以连接其子树的两条链
略微想了一下n≤3000n\le 3000n≤3000的dpi,jdp_{i,j}dpi,j以iii为根的子树用了iii及以上的祖先共jjj个,然后最大值dpdpdp,但是就是没写出来。。。
回归正题,没错一个节点确实可以将子树中的两个序列拼接起来
且处理完父亲节点后就没必要管儿子节点了
所以用堆来维护,点和点的子树所能组成的序列,从下往上合并,每次相当于是子树全合并给父亲,然后父亲选最长的两个子序列进行拼接合并
启发式合并即可
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;
}