【USACO】 奶牛政坛

【题目链接】

点击打开链接

【算法】

tarjan算法求LCA

【代码】

#include<bits/stdc++.h>
#define MAXN 200010
#pragma GOC optimize("O2")
 
using namespace std;
 
int n,k,i,p,fa,q;
int a[MAXN],visit[MAXN],parent[MAXN],maxn[MAXN],
    depth[MAXN],x[MAXN],y[MAXN],z[MAXN],ans[MAXN];
vector<int> son[MAXN],vec[MAXN];
 
int find(int x) {
    if (parent[x] == x) return x;
    parent[x] = find(parent[x]);
    return parent[x];
} 
 
void dfs(int dep,int d) {
    int i;
    depth[dep] = d;
    for (i = ; i < son[dep].size(); i++)
        dfs(son[dep][i],d+);
}
 
void tarjan(int u) {
    int i;
    parent[u] = u;
    visit[u] = ;
    for (i = ; i < vec[u].size(); i++) {
        if (y[vec[u][i]] == u && visit[x[vec[u][i]]]) z[vec[u][i]] = find(x[vec[u][i]]);
        if (x[vec[u][i]] == u && visit[y[vec[u][i]]]) z[vec[u][i]] = find(y[vec[u][i]]);
    }
    for (i = ; i < son[u].size(); i++) {
        if (!visit[son[u][i]]) {
            tarjan(son[u][i]);
            parent[son[u][i]] = u;
        }
    }
}
 
int main() {
 
    scanf("%d%d",&n,&k);
    for (i = ; i <= n; i++) {
        scanf("%d%d",&a[i],&p);
        if (p != ) son[p].push_back(i);
        else fa = i;
    }
 
    dfs(fa,);
 
    for (i = ; i <= n; i++) {
        if (depth[i] > depth[maxn[a[i]]])
            maxn[a[i]] = i;
    }
 
    for (i = ; i <= n; i++) {
        if (maxn[a[i]] != i) {
            ++q;
            x[q] = maxn[a[i]];
            y[q] = i;
            vec[maxn[a[i]]].push_back(q);
            vec[i].push_back(q);
        }
    }
 
    tarjan(fa);
 
    for (i = ; i <= q; i++) {
        ans[a[x[i]]] = max(ans[a[x[i]]],depth[x[i]] + depth[y[i]] -  * depth[z[i]]);
    }
    for (i = ; i <= k; i++) printf("%d\n",ans[i]);
 
    return ;
 
}