codeforces 321C 点分治好题

题意：给定一棵树，让你对其进行染色，使得任意颜色相同的两个点之间的路径上存在等级更高的颜色。

思路：我们考虑点分树的结构，点分树上的每个节点都对应于一个边的集合（经过这个节点的所有边）。

点分树上的父子关系，一棵点分树上任意一个节点，以他为根的子树上的所有点是这个节点路径集合的端点集合。

有了这些结论，这道题就变成了对点分树进行染色，每层都染成一个颜色。因为点分树的层数最多为logn，所以一定存在合法的染色方案。

#include <cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn = 100000 + 5;
const int maxm = 200000 + 5;
int he[maxn],ne[maxm],ver[maxm],tot;
void add( int x,int y ){
    ver[++tot] = y;
    ne[tot] = he[x];
    he[x] = tot;
}
int root,  // 当前树的重心
vis[maxn], // 标记点是否作为重心出现过
n,        // 树的总点数
sz[maxn],
g[maxn];
void get_sz( int x,int f ){
    sz[x] = 1;
    g[x] = 0;
    for( int cure = he[x];cure;cure = ne[cure] ){
        int y  = ver[cure];
        if( y == f || vis[y] ) continue;
        get_sz( y,x );
        g[x] = max( g[x],sz[y] );
        sz[x] += sz[y];
    }
}
int find_root( int x,int f ){
    //g[x] = max( g[x],sz[root] - sz[x] );
    int minx = n;
    int re = x;
    for( int cure = he[x];cure;cure = ne[cure] ){
        int y  = ver[cure];
        if( vis[y] || y == f ) continue;
        int tmp = find_root( y,x );
        if( g[tmp] < minx ){
            re = tmp,minx = g[tmp];
        }
    }
    g[x] = max( g[x],sz[root]-sz[x] );
    if( g[x] < minx ) re = x;
    return re;
}
void get_root( int& x ){
    get_sz( x,0 );
    root = x;
    x = find_root( x,0 );
}
int ans[maxn];
void dfs_div( int x,int c ){
    if(vis[x]) return;
    get_root(x);
    vis[x] = 1;
    ans[x] =c;     
    for( int cure = he[x];cure;cure = ne[cure] ){
        int y = ver[cure];
        if( vis[y] ) continue;
        dfs_div( y,c+1 );
    }
}
void init(int n){
    memset( he,0,sizeof(int)*(n+1) );
    memset( vis,0,sizeof(int)*(n+1) );
    tot = 1;
}
int main(){
    //freopen( "C:\\Users\\qqq\\Desktop\\987.txt","r",stdin );
    int x,y;
    scanf("%d",&n);
    init(n);
    for( int i = 1;i < n;i++ ){
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs_div(1,0);
    for( int i = 1;i <= n;i++ ){
        printf("%c ",ans[i]+'A');
    }
    puts("");
    return 0;
}