Garland CodeForces - 767C

http://codeforces.com/problemset/problem/767/C

找两个除根节点之外的点 将整棵树分为权值相等的三部分

两个点的关系 要么lca是两者其中一个 要么不是

对于第一种情况 在dfs时用栈保存权值为(2/3*tot)的点 如果遍历到某个权值为(1/3*tot)的点时战非空 则找到答案

对于第二种情况 对于遍历到的每棵子树 看它有多少棵子树含有权值为(1/3*tot)的点 超过一棵则找到答案

 

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;

struct node
{
    int v,next;
};

stack <int> stk;
node edge[maxn];
int first[maxn],sum[maxn],book[maxn];
int n,num,tot,root,flag,ans1,ans2;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfsI(int cur)
{
    int i,v;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        dfsI(v);
        sum[cur]+=sum[v];
    }
    if(sum[cur]==tot) book[cur]=cur;
}

void dfsII(int cur)
{
    int i,v;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        dfsII(v);
        if(book[cur]==0&&book[v]!=0) book[cur]=book[v];
    }
}

void dfsIII(int cur)
{
    int i,v,cnt,pre;
    if(cur!=root&&sum[cur]==2*tot){
        stk.push(cur);
    }
    cnt=0;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        dfsIII(v);
        if(!stk.empty()&&sum[v]==tot){
            flag=1,ans1=stk.top(),ans2=v;
        }
        if(book[v]!=0){
            if(cnt==0) cnt=1,pre=book[v];
            else{
                flag=1,ans1=pre,ans2=book[v];
            }
        }
    }
    if(cur!=root&&sum[cur]==2*tot){
        stk.pop();
    }
}

int main()
{
    int i,u;
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    num=0,tot=0;
    for(i=1;i<=n;i++){
        scanf("%d%d",&u,&sum[i]);
        if(u==0) root=i;
        else addedge(u,i);
        tot+=sum[i];
    }
    if(tot%3!=0) printf("-1\n");
    else{
        tot/=3;
        dfsI(root);
        dfsII(root);
        dfsIII(root);
        if(flag) printf("%d %d\n",ans1,ans2);
        else printf("-1\n");
    }

    return 0;
}