CF 735E

题目:

一棵树上有黑白两种颜色,一次染色能对相同颜色的区域取反,求几次染色可使得整棵树为同一种颜色。

PS: 还是刷不动F,心累。

题解:

缩点后求树的直径/2 。

不严谨证明:

把一棵树立起来,不断地对树根染色,这样就会发现只需要考虑最长的树链。同理,如果对树链上某一点着色,一定次数的操作后会等价于对树根染色,贪心地选取,这个点应该是最长树链的中点。

#include<iostream>
#include<vector>
#include<cstring>
using namespace std;

int n;
int maxlen=0;
int duan=0;
int arr[201000];
int in[201000];
bool vis[201000];
bool vistuan[201000];
vector<int> vec[201000];
vector<int> tuan[201000];

void dfs(int bit,int x,int color){
    if(vis[x]) return ;
    if(arr[x]!=color) return ;
    in[x]=bit;
    vis[x]=true;
    for(int i=0;i<vec[x].size();i++){
        dfs(bit,vec[x][i],color);
    }
}

void findlen(int x,int len){
    if(vistuan[x]) return ;
    vistuan[x]=true;
    if(len>maxlen){
        duan=x;
        maxlen=len;
    }
    for(int i=0;i<tuan[x].size();i++){
        findlen(tuan[x][i],len+1);
    }
}

int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>arr[i];
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        vec[a].push_back(b);
        vec[b].push_back(a);
    }
    int bit=0;
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            dfs(bit,i,arr[i]);
            bit++;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<vec[i].size();j++){
            int to=vec[i][j];
            if(in[i]!=in[to]){
                tuan[in[i]].push_back(in[to]);
            }
        }
    }
    findlen(1,1);
    maxlen=0;
    memset(vistuan,0,sizeof(vistuan));
    findlen(duan,1);
    cout<<maxlen/2<<endl;
}

阅读更多
个人分类: 树上操作
上一篇CF 735D
下一篇CF 742D
想对作者说点什么? 我来说一句

易语言模块易语言模块

2010年08月14日 529KB 下载

没有更多推荐了,返回首页

关闭
关闭