A10766 | Anton and Tree
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
Anton is growing a tree in his garden. In case you forgot, the tree is a connected acyclic undirected graph.
There are $n$ vertices in the tree, each of them is painted black or white. Anton doesn't like multicolored trees, so he wants to change the tree such that all vertices have the same color (black or white).
To change the colors Anton can use only operations of one type. We denote it as $paint(v)$ , where $v$ is some vertex of the tree. This operation changes the color of all vertices $u$ such that all vertices on the shortest path from $v$ to $u$ have the same color (including $v$ and $u$ ). For example, consider the tree
and apply operation $paint(3)$ to get the following:
Anton is interested in the minimum number of operation he needs to perform in order to make the colors of all vertices equal.
There are $n$ vertices in the tree, each of them is painted black or white. Anton doesn't like multicolored trees, so he wants to change the tree such that all vertices have the same color (black or white).
To change the colors Anton can use only operations of one type. We denote it as $paint(v)$ , where $v$ is some vertex of the tree. This operation changes the color of all vertices $u$ such that all vertices on the shortest path from $v$ to $u$ have the same color (including $v$ and $u$ ). For example, consider the tree
and apply operation $paint(3)$ to get the following:
Anton is interested in the minimum number of operation he needs to perform in order to make the colors of all vertices equal.
输入格式
The first line of the input contains a single integer $n$ ( $1<=n<=200000$ ) — the number of vertices in the tree.
The second line contains $n$ integers $color_{i}$ ( $0<=color_{i}<=1$ ) — colors of the vertices. $color_{i}=0$ means that the $i$ -th vertex is initially painted white, while $color_{i}=1$ means it's initially painted black.
Then follow $n-1$ line, each of them contains a pair of integers $u_{i}$ and $v_{i}$ ( $1<=u_{i},v_{i}<=n,u_{i}≠v_{i}$ ) — indices of vertices connected by the corresponding edge. It's guaranteed that all pairs $(u_{i},v_{i})$ are distinct, i.e. there are no multiple edges.
The second line contains $n$ integers $color_{i}$ ( $0<=color_{i}<=1$ ) — colors of the vertices. $color_{i}=0$ means that the $i$ -th vertex is initially painted white, while $color_{i}=1$ means it's initially painted black.
Then follow $n-1$ line, each of them contains a pair of integers $u_{i}$ and $v_{i}$ ( $1<=u_{i},v_{i}<=n,u_{i}≠v_{i}$ ) — indices of vertices connected by the corresponding edge. It's guaranteed that all pairs $(u_{i},v_{i})$ are distinct, i.e. there are no multiple edges.
输出格式
Print one integer — the minimum number of operations Anton has to apply in order to make all vertices of the tree black or all vertices of the tree white.
输入输出样例
输入 #1
11 0 0 0 1 1 0 1 0 0 1 1 1 2 1 3 2 4 2 5 5 6 5 7 3 8 3 9 3 10 9 11
输出 #1
2
输入 #2
4 0 0 0 0 1 2 2 3 3 4
输出 #2
0
In the first sample, the tree is the same as on the picture. If we first apply operation $paint(3)$ and then apply $paint(6)$ , the tree will become completely black, so the answer is $2$ .
In the second sample, the tree is already white, so there is no need to apply any operations and the answer is $0$ .
In the second sample, the tree is already white, so there is no need to apply any operations and the answer is $0$ .
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted