F - Garland CodeForces - 767C（dfs）

题目点击打开链接

记得别用G++

超时的一个鬼

这个题 用一个vector来存储树，

找到之后就将当前的t【u】清零，之后便再不用走过这个路径

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<vector>

using namespace std;

const int maxn=1000000+5;
int t[maxn];

vector<int> g[maxn];
int n,root;
int sum;
int v1,v2;
void dfs(int u)
{
    if(v1&&v2)
        return ;
    int l=g[u].size();
    for(int i=0;i<l;i++)
    {
        int v=g[u][i];
        dfs(v);
       t[u]+=t[v];
    }
    if(t[u]==sum/3&&u!=root)
    {
        if(!v1) {v1=u;t[u]=0;}
        else if(!v2){v2=u;t[u]=0;return ;}
    }
}
int main()
{

    cin>>n;
    memset(t,0,sizeof(t));

    for(int i=0;i<n;i++)
        g[i].clear();
    for(int i=1;i<=n;i++)
    {
       int x,y;
       scanf("%d%d",&x,&y);
       sum+=y;
       t[i]=y;
       if(!x)
        root=i;
       g[x].push_back(i);
    }
    if(sum%3)
     cout<<"-1"<<endl;
     else
     {
         dfs(root);
         if(v1&&v2)
            cout<<v1<<" "<<v2<<endl;
         else
         cout<<"-1"<<endl;
     }

    return 0;
}

dxcf