芭芭拉冲鸭~ 树的直径

https://ac.nowcoder.com/acm/contest/8755/H

思路：不同类型的点对他进行建边，等于是将原来的一棵树断开了一些边，成了一片森林，那么对每棵树求一下直径，维护max即可。可以用并查集维护每棵树内的结点。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=1e6+50;
int f[N];
int find(int x){
    return f[x]==x?f[x]:f[x]=find(f[x]);
}
int r,ans;
char c[N];
bool vis[N];
int dis[N];
vector<int> e[N];
vector<int> Q[N];
void bfs(int root){
    queue<int> q;
    q.push(root);
    vis[root]=1;
    dis[root]=0;
    while(!q.empty()){
        int now=q.front();
        q.pop();
        for(auto it:e[now]){
            if(!vis[it]){
                vis[it]=1;
                dis[it]=dis[now]+1;
                if(ans<dis[it]) ans=dis[it],r=it;
                q.push(it);
            }
        }
    }
}
int sum;
void solve(int s){
    ans=0;
    for(auto z:Q[s]){
        vis[z]=0;
        dis[z]=0;
    }
    bfs(s);
    ans=0;
    for(auto z:Q[s]){
        vis[z]=0;
        dis[z]=0;
    }
    bfs(r);
    sum=max(sum,ans);
}
int main(){
    int n;cin>>n;
    for(int i=1;i<=n;i++) cin>>c[i],f[i]=i;
    for(int i=1;i<n;i++){
        int x,y;cin>>x>>y;
        if(c[x]^c[y]){
            f[find(x)]=find(y);
            e[x].push_back(y);
            e[y].push_back(x);
        }
    }
    for(int i=1;i<=n;i++) Q[find(i)].push_back(i);
    for(int i=1;i<=n;i++){
        if(f[i]=find(i)){
            solve(i);
        }
    }
    cout<<sum;
    return 0;
}