求树上最长美丽路径的长度

fa602936 程序题 共 个子题

题目信息

时间限制:1.0 s 内存限制:512.0 MB

题面描述

小杨有一棵包含n个节点的树,节点从1到n编号,每个节点要么是白色,要么是黑色。 树上的一条简单路径(不经过重复节点)是美丽路径当且仅当路径上相邻节点的颜色均不相同。例如下图中,节点1和节点4是黑色,其余节点为白色:路径2-1-3-4是美丽路径,路径2-1-3-5不是美丽路径(相邻节点3和5颜色相同)。 树示例 路径的长度为路径包含的节点数量,请你求出最长的美丽路径的长度。

输入格式

第一行包含一个正整数n,代表节点数量。 第二行包含n个整数$c_1,c_2,\dots,c_n$,代表每个节点的颜色:$c_i=0$表示节点i为白色,$c_i=1$表示节点i为黑色。 之后$n-1$行,每行包含两个正整数$u_i,v_i$,代表存在一条连接节点$u_i$和$v_i$的边。

输出格式

输出一个整数,代表最长美丽路径的长度。

样例1

输入

5
1 0 0 1 0
1 2
3 5
4 3
1 3

输出

4

样例2

输入

5
0 0 0 0 0
1 2
2 3
3 4
4 5

输出

1

数据范围

子任务编号 数据点占比 n 特殊条件
1 30% ≤ 1000 树的形态是一条链
2 30% ≤ 1000 无
3 40% ≤ $10^5$ 无

对于全部数据,保证$1 \leq n \leq 10^5$,$0 \leq c_i \leq 1$,输入数据为合法的树结构。

本题为编程题,提交代码并运行通过测试点后判分。

正在加载题目数据…
已答 /
子题
该题型需在试卷/作业中作答,单题练习暂不支持
答案解析
提交记录
共 条 仅展示自由练习提交
正在加载提交记录…
讨论
发布
登录后参与讨论 立即登录
发布讨论需先绑定并验证手机号 去绑定手机号
加载中…
我
我
回复 @
回复 @
发送
还没有人讨论,来发表第一条吧