Garland CodeForces

  题意就是给你一棵树,然后要求你将这棵树三等分,即选2个节点,将这2个节点与其父节点的边切掉,不能选根节点。

  做法就是,选判断一下总点权值tmp能不被三整出,能的话则可能可以三等分,否则的话肯定不能三等分。然后还要注意一下可能将树三等分可以切的点的组合(2个点为一个组合)有多个只要一种即可 ,从root开始往下dfs,回溯的时候更新一下节点对应的子树的点权值,若满足sum[now]==tmp,说明该节点是可以切去的,切掉了之后sum【now】要清0,因为切掉之和子树now对其父节点的贡献就是0了。ps:这题千万能用cin读入。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
vector<int>g[maxn];
ll sum[maxn];
int val[maxn],ans[5],tmp,num,root,n;
void dfs(int now)
{

    sum[now]=val[now];
    int sz=g[now].size();
    for(int i=0;i<sz;i++)
    {
        int v=g[now][i];
        dfs(v);
        sum[now]+=sum[v];
    }
    if(root!=now&&sum[now]==tmp&&num!=2)
    {
        ans[num++]=now;
        sum[now]=0;
    }
}
int main()
{

    scanf("%d",&n);
    for(int i=1;i<=n;i++) g[i].clear();
    tmp=num=0;
    for(int i=1;i<=n;i++)
    {
        int u,v=i;
        scanf("%d %d",&u,&(val[i]));
        tmp+=val[i];
        if(u==0)
        {
            root=i;
            continue;
        }
        g[u].push_back(i);
    }
    if(tmp%3!=0)
    {
        cout<<-1<<endl;
        return 0;
    }
    tmp/=3;
    dfs(root);
    if(num==2)
        cout<<ans[0]<<" "<<ans[1]<<endl;
    else
        cout<<-1<<endl;
    return 0;
}
原文地址:https://www.cnblogs.com/eason9906/p/11754893.html