E. Anton and Tree[树的直径]

E. Anton and Tree

转化的题意: 一张黑白相间的树,一次操作可以把一片联通的白色顶点变成黑色,黑色变成白色.问至少需要多少次操作让图变成一种颜色

思路: 树的直径为d 答案为 (d+1)/2 

证明: 首先,至少为 (d+1)/2 , 现在需要证明这个答案能成立. 每次取中间的点,因为右边的直径永远会比右边点的侧枝要长,所以一定可以减完.左边同理

#include
#define PI acos(-1.0)
#define pb push_back
#define F first
#define S second
using namespace std;
typedef long long ll;
const int N=2e5+5;
const int MOD=1e9+7;
ll a[N],sum[N];
ll c[N];
int par[N];
vector edge[N];
vector now[N];
bool vis[N];
int cnt=0;
int mp[N];
void bfs(int u){
    queue  q;
    q.push(u);
    vis[u]=true;
    ++cnt;
    while(!q.empty()){
        int v=q.front();
        mp[v]=cnt;
        q.pop();
        for(auto t : edge[v]){
            if(vis[t]||c[u]!=c[t])  continue;
            vis[t]=true;
            q.push(t);
        }
    }
}
ll dis[N];
void dfs(int u,int f){
    par[u]=f;
//    cout <<"u="<=mx) mx=dis[i],s=i;
//    cout <<"s="<=mx) mx=dis[i],ed=i;
//    cout <<"ed="<"<>n ;
    for(int i=1;i<=n;i++)   cin >> c[i];
    for(int i=1;i>u>>v;
        edge[u].pb(v);
        edge[v].pb(u);
    }
    ll res=0;
    for(int i=1;i<=n;i++){
        if(c[i]==0&&!vis[i]) res++,bfs(i);
    }
    memset(vis,false,sizeof vis);
    ll ans=0;
    for(int i=1;i<=n;i++){
        if(c[i]==1&&!vis[i]) ans++,bfs(i);
    }

    for(int u=1;u<=n;u++){
        for(auto v : edge[u]){
            if(mp[u]==mp[v])    continue;
            now[mp[u]].pb(mp[v]);
        }
    }

////    for(int i=1;i<=n;i++)   cout << "mp["<