Garland

做法

• 1.删除掉两条边所以会分成相同的三部分 ---> 先判断树上的值是否能被3整除
• 2.求出三部分的值是什么
• 3.对树跑一边dfs，记录能分成这个值的边
• 4.如果由三部分则输出答案，否则输出

代码

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp(aa,bb) make_pair(aa,bb)
#define _for(i,b) for(int i=(0);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,b,a) for(int i=(b);i>=(a);i--)
#define mst(abc,bca) memset(abc,bca,sizeof abc)
#define X first
#define Y second
#define lowbit(a) (a&(-a))
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
typedef long double ld;
const int N=1e6+10;
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-6;
const double PI=acos(-1.0);

int n,root,w[N],t[N],sum;
vector<int> g[N],ans;

void dfs(int u,int fa){
    t[u]=w[u];
    for(auto v:g[u]){
        if(v==fa) continue;
        dfs(v,u);
        t[u]+=t[v];
    }
    if(t[u]==sum) ans.pb(u),t[u]=0;
}

void solve(){
    cin>>n;
    rep(i,1,n){
        int fa;
        cin>>fa>>w[i];
        if(fa){
            g[fa].pb(i);
            g[i].pb(fa);
        } 
        else root=i;
        sum+=w[i];
    }
    if(sum%3){
        cout<<"-1\n";
        return;
    }
    sum/=3;
    dfs(root,0);
    if(ans.size()<=2) cout<<"-1\n";
    else cout<<ans[0]<<" "<<ans[1]<<"\n"; 
}


int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
#ifdef DEBUG
    freopen("F:/laji/1.in", "r", stdin);
//    freopen("F:/laji/2.out", "w", stdout);
#endif
//    int t;cin>>t;while(t--)
    solve();
    return 0;
}