bzoj1776

点分治/贪心

对于点分治的理解不够深刻...点分治能统计树上每个点对的信息，那么这里就是统计同种颜色点对之间的最大距离，自然可以用点分

然后点分，每次统计最大距离，但是略微卡常...

还有一种贪心的方法，每种颜色必然选以某点为根最深的节点，计算出最深的节点，然后dfs，看每种颜色，然后和最深的节点计算距离

#include<bits/stdc++.h>
using namespace std;
const int N = ;
int n, top, root, k, top1;
vector<int> G[N];
int size[N], mx[N], mark[N], ans[N], st[N], st1[N], c[N], a[N], mx_deep[N], now_deep[N], vis[N], vis1[N];
namespace IO
{
    const int Maxlen = N * ;
    char buf[Maxlen], *C = buf;
    int Len;
    inline void read_in()
    {
        Len = fread(C, , Maxlen, stdin);
        buf[Len] = '\0';
    }
    inline void fread(int &x)
    {
        x = ;
        int f = ;
        while (*C < '' || '' < *C) { if(*C == '-') f = -; ++C; }
        while ('' <= *C && *C <= '') x = (x << ) + (x << ) + *C - '', ++C;
        x *= f;
    }
    inline void read(int &x)
    {
        x = ;
        int f = ; char c = getchar();
        while(c < '' || c > '') { if(c == '-') f = -; c = getchar(); }
        while(c >= '' && c <= '') { x = (x << ) + (x << ) + c - ''; c = getchar(); }
        x *= f;
    }
    inline void read(long long &x)
    {
        x = ;
        long long f = ; char c = getchar();
        while(c < '' || c > '') { if(c == '-') f = -; c = getchar(); }
        while(c >= '' && c <= '') { x = (x << 1ll) + (x << 3ll) + c - ''; c = getchar(); }
        x *= f;
    }
} using namespace IO;
void findroot(int u, int last, int tot)
{
    size[u] = ;
    mx[u] = ;
    for(int i = ; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        if(v == last || mark[v]) continue;
        findroot(v, u, tot);
        size[u] += size[v];
        if(size[v] > mx[u]) mx[u] = size[v];
    }
    mx[u] = max(mx[u], tot - mx[u]);
    if(mx[u] < mx[root]) root = u;
}
void dfs(int u, int last, int deep)
{
    if(mx_deep[a[u]] == -) st[++top] = a[u];
    mx_deep[a[u]] = max(mx_deep[a[u]], deep);
    for(int i = ; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        if(v == last || mark[v]) continue;
        dfs(v, u, deep + );
    }
}
int get_size(int u, int last)
{
    int ret = ;
    for(int i = ; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        if(v == last || mark[v]) continue;
        ret += get_size(v, u);
    }
    return ret;
}
void divide(int u)
{
    root = ;
    findroot(u, , get_size(u, ));
    mark[root] = ;
    now_deep[a[root]] = ;
    st1[++top1] = a[root];
    for(int i = ; i < G[root].size(); ++i)
    {
        int v = G[root][i];
        if(mark[v]) continue;
        dfs(v, root, );
        if(mx_deep[a[root]] == -) st[++top] = a[root], mx_deep[a[root]] = ;
        for(int j = top; j; --j) if(now_deep[st[j]] != -) ans[st[j]] = max(ans[st[j]], now_deep[st[j]] + mx_deep[st[j]]);
        while(top)
        {
            if(now_deep[st[top]] == -) st1[++top1] = st[top];
            now_deep[st[top]] = max(now_deep[st[top]], mx_deep[st[top]]);
            mx_deep[st[top--]] = -;
        }
    }
    while(top1) now_deep[st1[top1--]] = -;
    for(int i = , now = root; i < G[now].size(); ++i)
    {
        int v = G[now][i];
        if(mark[v]) continue;
        divide(v);
    }
}
int main()
{
    memset(now_deep, -, sizeof(now_deep));
    memset(mx_deep, -, sizeof(mx_deep));
    read_in();
    fread(n);
    fread(k);
    for(int i = ; i <= n; ++i)
    {
        int u;
        fread(a[i]);
        fread(u);
        if(u) { G[u].push_back(i); G[i].push_back(u); }
    }
    mx[] =  << ;
    divide();
    for(int i = ; i <= k; ++i) printf("%d\n", ans[i]);
    return ;
}