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

来源：互联网发布：信息系统数据保密协议编辑：程序博客网时间：2024/05/03 13:44

题目链接：http://codeforces.com/contest/767/problem/C

题解：类似于提着一串葡萄，用剪刀剪两条藤，葡萄分成了三串。问怎样剪才能使三串葡萄的质量相等。

首先要做的就是统计葡萄的总质量tot。之后就是找到两子串质量为（tot/3）的葡萄（如果除不尽，则必定找不到），那么剩下的就是dfs搜索了。

我一开始的做法是先建一棵记录子树质量和的树，然后再从上往下dfs，如果找到了，就把它剪掉。后来发现被剪掉的那一串可能就含有两串质量为（tot/3）的葡萄（这里质量可为负数），所以这种方法行不通。

那么正确的做法是先深入到最底层（叶子结点），然后再从下往上搜索，这个过程与建树的过程是一致的。所以可以将两个过程合并在一起。

代码如下：

#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<queue>#include<vector>#include<map>#include<string>#include<set>#define LL long long#define MAX(a,b) (a>b?a:b)#define MIN(a,b) (a<b?a:b)#define INF 0x7fffffff#define LNF ((1LL<<62)-1)#define maxn 200010using namespace std;int n, tot = 0,cnt = 0, sum[1000005], vis[1000005];vector<int>child[1000005];int ans[5];int build(int rt){    vis[rt] = 1;    int m = child[rt].size();    for(int i = 0; i<m; i++)    {        if(!vis[child[rt][i]])        sum[rt] += build(child[rt][i]);        if(cnt==2)            return 0;    }    if(sum[rt]==tot)    {        ans[cnt++] = rt;        return 0;    }    else return sum[rt];}int main(){    int rt,v;    scanf("%d",&n);    for(int i = 1; i<=n; i++)    {        scanf("%d%d",&v,&sum[i]);        if(v)        child[i].push_back(v);        child[v].push_back(i);        tot += sum[i];        if(!v) rt = i, vis[rt] = 1;    }    if(tot%3)    {        puts("-1");        return 0;    }    tot /= 3;    int m = child[rt].size();    for(int i = 0; i<m; i++)    {        build(child[rt][i]);        if(cnt==2) break;    }    if(cnt==2)        printf("%d %d\n",ans[0],ans[1]);    else        puts("-1");}

0 0