CodeForces 767C Garland【DFS+vector】

Source：CodeForces 767C

  题意：题目给出一个根节点已经确定的树，每个节点上有个权值，这个权值可以为负数，0，正数（可以为负数！），然后求问能不能找到两个节点，分别将从这两个节点到其父节点的边砍掉，由此形成的三个非空部分的权值和相同？

  思路： 首先分成三个部分权值和相同，说明要能够分出来这个树一定满足权值总和为3的倍数。其次容易想到我们需要计算以每个节点为根的子树的权值和，这涉及到树的遍历，一般是用dfs进行。再者，从贪心角度看，递归dfs时，结果是从叶节点往上依次计算出来的，所以一旦发现计算到某个节点其下权值和为sum/3，则砍掉这一子树一定不亏，砍掉后直接向上层函数返回0即可。。

  注意： 用dfs返回权值；同时注意砍掉子树的根节点不能为原本根节点；因为权值可以为负数所以两部分可以在同一个子树上面，用dfs返回权值的模式巧妙地避开了一些复杂的讨论（毕竟cf的题数据都很卡人，思路一定要严密）。

代码如下：

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
const int max_n=1e6+5;

int n,sum,t[max_n],f,root;
vector <int> g[max_n],ans;

int dfs(int v)
{
    int tem=t[v];
    for(int i=0;i<g[v].size();i++)
        tem+=dfs(g[v][i]);
    if(tem==sum/3 && v!=root)
    {
        ans.push_back(v);
        tem=0;            //砍掉这个子树
    }
    return tem;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&f,&t[i]);
        if (f) g[f].push_back(i);
        else root=i;
        sum+=t[i];
    }
    if(sum%3) printf("-1\n");
    else
    {
        dfs(root);
        if(ans.size()<2)printf("-1\n");
        else
        printf("%d %d",ans[0],ans[1]);
    }
    return 0;
}