Codeforces Round #379 (Div. 2) E Anton and Tree

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 aspaint(v), where v is some vertex of the tree. This operation changes the color of all verticesu such that all vertices on the shortest path fromv to u have the same color (includingv and u). For example, consider the tree

Codeforces Round #379 (Div. 2) E Anton and Tree_第1张图片

and apply operation paint(3) to get the following:

Codeforces Round #379 (Div. 2) E Anton and Tree_第2张图片

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 thei-th vertex is initially painted white, whilecolori = 1 means it's initially painted black.

Then follow n - 1 line, each of them contains a pair of integersui andvi (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 operationpaint(3) and then apply paint(6), the tree will become completely black, so the answer is2.

In the second sample, the tree is already white, so there is no need to apply any operations and the answer is0.



好题!考察了树的直径的作用。题意中的paint(v)就是把所有的与v同色的并从这个点到v不经过其他颜色的点都涂成一种颜色,包括v;

这样缩点是必然的,用dfs缩点后重构树(可证:重构图必然是树)

然后因为paint(v)对所有点有效,我们很容易看出树的直径/2就是答案



#include 

using namespace std;
const int maxn=4e5+5;
int num[maxn],vis[maxn],headpre[maxn],headafter[maxn];
struct node
{
    int from,to,next,v;
};
int cnt=0,edgeprenum=1,edgeafternum=1,maxnum=0,maxid;
node edgepre[maxn],edgeafter[maxn];
void addedgepre(int u,int v)
{
    edgepre[edgeprenum].from=u;
    edgepre[edgeprenum].to=v;
    edgepre[edgeprenum].next=headpre[u];
    headpre[u]=edgeprenum++;
}
void addedgeafter(int u,int v)
{
    edgeafter[edgeafternum].from=u;
    edgeafter[edgeafternum].to=v;
    edgeafter[edgeafternum].next=headafter[u];
    headafter[u]=edgeafternum++;
}

int dfs(int now)
{
    vis[now]=cnt;
    for(int i=headpre[now];i!=-1;i=edgepre[i].next)
    {
        int v=edgepre[i].to;
        if(!vis[v]&&num[now]==num[v])
            dfs(v);
    }
}
int bfs(int now)
{
    memset(vis,0,sizeof(vis));
    queueque;
    que.push(now);
    maxnum=0;
    vis[now]=1;
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        if(maxnum





你可能感兴趣的:(Codeforces Round #379 (Div. 2) E Anton and Tree)