E. Anton and Tree(缩点+树直径模板)

本篇介绍了一种算法问题,旨在找到使树形图所有节点变为同一颜色所需的最少操作次数。通过使用图论中的概念,如缩点和求树的直径,来解决这一挑战。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

E. Anton and Tree
time limit per test
3 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

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.

Input

The first line of the input contains a single integer n (1 ≤ n ≤ 200 000) — the number of vertices in the tree.

The second line contains n integers colori (0 ≤ colori ≤ 1) — colors of the vertices. colori = 0 means that the i-th vertex is initially painted white, while colori = 1 means it's initially painted black.

Then follow n - 1 line, each of them contains a pair of integers ui and vi (1 ≤ ui, vi ≤ n, ui ≠ vi) — indices of vertices connected by the corresponding edge. It's guaranteed that all pairs (ui, vi) are distinct, i.e. there are no multiple edges.

Output

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.

Examples
input
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
output
2
input
4
0 0 0 0
1 2
2 3
3 4
output
0
Note

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

题意:求得最小操作次数使得,一个只有黑白两种颜色的无向图变成一种颜色

解题思路:用了缩点和求树的直径,将相同颜色临近的点缩成一个颜色,然后关于求树的直径,先任意一点求BFS获得最长的那个点,从他再BFS一边获得的最长点就是树的直径,求出来后,可以近似的处理成一个树,然后就是树的直径向上取整


#include<iostream>  
#include<cstdio>
#include<stdio.h>
#include<cstring>  
#include<cstdio>  
#include<climits>  
#include<cmath> 
#include<vector>
#include <bitset>
#include<algorithm>  
#include <queue>
#include<map>
using namespace std;
int a[200005], n;
vector<int> tree[200005];
pair<int, int> temp;
void dfs(int root, int son, int deep)
{
	int i;
	pair<int, int> t;
	t = make_pair(deep, son);
	temp = max(temp, t);
	for (i = 0; i < tree[son].size(); i++)
	{
		int u = tree[son][i];
		if (root ==u)
		{
			continue;
		}
		if (a[u] == a[son])
		{
			dfs(son, u, deep);
		}
		else
		{
			dfs(son, u, deep + 1);
		}
	}
}
int main()
{
	int i, x, y;
	pair<int, int> s;
	cin >> n;
	for (i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	for (i = 1; i < n; i++)
	{
		cin >> x >> y;
		tree[x].push_back(y);
		tree[y].push_back(x);
	}
	temp = make_pair(0, 0);
	dfs(-1, 1, 0);
	s = temp;
	temp = make_pair(0, 0);
	dfs(-1, s.second, 0);
	cout << (temp.first + 1) / 2 << endl;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值