Codeforces Round #398 (Div. 2) C. Garland

Codeforces Round #398 (Div. 2)

C. Garland

题目链接

题目大意：

    给一棵树以及它的根节点，每个节点存一个值，题目中这个值是温度，Dima拿住这个根节点，他想把整个树剪成三段，使得三段上的温度和相等。(拿住根节点是想让根节点在中间的一段)

---

    直接从根节点开始DFS，每次更新子树上的温度和，如果温度和==总和/3，记录答案，除去该子树，tree[x]=0(以后DFS不再使用)    结束DFS后再判断根节点的和是否==总和/3，如果不是输出-1

---

#include<cstdio>using namespace std;int n,num[1000009],summ,san,ans[10],aa,root;struct E{    int next,to;}edge[2000060];int head[2000060],edge_num;int tree[1000009];void addedge(int x,int y){    edge[++edge_num].next=head[x];    edge[edge_num].to=y;    head[x]=edge_num;}void DFS(int x){    int i;    for(i=head[x];i;i=edge[i].next){        DFS(edge[i].to);        tree[x]+=tree[edge[i].to];    }    if(tree[x]==san&&aa<=1&&x!=root) tree[x]=0,ans[aa++]=x;}int main(){    freopen("Garland.in","r",stdin);    ans[0]=ans[1]=-1;    scanf("%d",&n);    int i;    for(i=1;i<=n;i++){        int a,b;        scanf("%d%d",&a,&b);        tree[i]=b;        if(a==0) root=i;        else addedge(a,i);        summ+=b;    }    if(summ%3!=0){        printf("-1");return 0;    }    san=summ/3;    DFS(root);    if(ans[0]!=-1&&ans[1]!=-1 && tree[root]==san){        printf("%d %d",ans[1],ans[0]);        return 0;    }    printf("-1");    return 0;}