概述
树数树
- description
- solution
- code
description
【题目描述】
牛牛有一棵 n 个点的有根树,根为 1。
我们称一个长度为 m 的序列 a 是好的,当且仅当:
•
∀
????
∈
(
1
,
????
]
∀????∈(1, ????]
∀i∈(1,m],
a
i
a_i
ai为
a
i
−
1
a_{i-1}
ai−1的祖先或
a
i
−
1
a_{i-1}
ai−1是
a
i
a_i
ai 的祖先
•
∀
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 ≤ 3000 nle 3000 n≤3000的 d p i , j dp_{i,j} dpi,j以 i i i为根的子树用了 i i i及以上的祖先共 j j j个,然后最大值 d p dp 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( "%dn", q[id[1]].top() );
}
return 0;
}
最后
以上就是甜美睫毛为你收集整理的[NowCoder牛客]2021NOIP提高组模拟赛第二场T3——树数树(启发式合并堆)descriptionsolutioncode的全部内容,希望文章能够帮你解决[NowCoder牛客]2021NOIP提高组模拟赛第二场T3——树数树(启发式合并堆)descriptionsolutioncode所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复