bzoj1776[Usaco2010 Hol]cowpol 奶牛政坛

来源：互联网发布：网络运营需要什么技术编辑：程序博客网时间：2024/05/19 17:10

现在才开始刷gold，我真是水的不行了。。

分析：这道题，，，我一开始想到了正解，但是以为是n^2的直接放弃了。。想半天没想法，，结果看题解之后一脸懵逼，之后才发现原来我复杂度算错了。。你枚举每个政党内所有点与最大深度的那个点，其实是一遍，所有的政党，不就O(n)..然后lca xjb上。。预处理放在dfs外面居然会wa。。一脸懵逼。

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,a,b) for(int i=a;i>=b;i--)using namespace std;int n,m;const int N=5e5+5;int head[N],go[N],next[N],fa[N][22];int dep[N],ans[N],a[N];int tot;bool vis[N];struct node{    int dep,id;}b[N];inline void add(int x,int y){    go[++tot]=y;    next[tot]=head[x];    head[x]=tot;}inline void dfs(int x){    vis[x]=1;    int i=head[x];    while (i)    {        int v=go[i];        if (!vis[v])        {            dep[v]=dep[x]+1;            //fa[v][0]=x;            dfs(v);        }        i=next[i];    }}inline int Lca(int x,int y){    if (dep[x]<dep[y])swap(x,y);    fd(i,20,0)    if(dep[fa[x][i]]>=dep[y])x=fa[x][i];    if (x==y)return x;    fd(i,20,0)    if (fa[x][i]!=fa[y][i])    {        x=fa[x][i];        y=fa[y][i];    }    return fa[x][0];}int main(){    scanf("%d%d",&n,&m);    int root=0;    fo(i,1,n)    {        int x,y;        scanf("%d%d",&y,&x);        a[i]=y;        fa[i][0]=x;        a[i]=y;        if (x==0)        {            root=i;            continue;        }        add(i,x);        add(x,i);    }    dep[root]=1;    dfs(root);    //fo(i,1,n)printf("%d ",dep[i]);    fo(i,1,n)    {        if (b[a[i]].dep<dep[i])        {            b[a[i]].dep=dep[i];            b[a[i]].id=i;        }    }    //fo(i,1,m)printf("%d %d\n",b[i].id,b[i].dep);    //return 0;    fo(i,1,n)    fo(j,1,20)    fa[i][j]=fa[fa[i][j-1]][j-1];    fo(i,1,n)    {        if (b[a[i]].id==i)continue;        int x=i,y=b[a[i]].id;        int lca=Lca(x,y);        //printf("%d %d %d\n",x,y,lca);        //printf("%d %d %d\n",dep[x],dep[y],dep[lca]);        ans[a[i]]=max(ans[a[i]],dep[x]+dep[y]-2*dep[lca]);    }    fo(i,1,m)printf("%d\n",ans[i]);}

0 0