VK Cup 2015 - Round 2 B. Work Group

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=200005;
const int mod=1000000007;
int n,x;
int w[maxn];
ll dp[maxn][2];
vector<int> s[maxn];

void dfs(int u)
{
    dp[u][0]=0;dp[u][1]=-(1LL<<60);
    for(int i=0;i<s[u].size();i++)
    {
        int v=s[u][i];
        dfs(v);
        ll t0=dp[u][0],t1=dp[u][1];
        t0=max(t0,dp[u][0]+dp[v][0]);
        t0=max(t0,dp[u][1]+dp[v][1]);
        t1=max(t1,dp[u][1]+dp[v][0]);
        t1=max(t1,dp[u][0]+dp[v][1]);
        dp[u][0]=t0;
        dp[u][1]=t1;
    }
    dp[u][1]=max(dp[u][1],dp[u][0]+w[u]);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&w[i]);
        if(x==-1) continue;
        s[x].push_back(i);
    }
    dfs(1);
    printf("%lld\n",max(dp[1][0],dp[1][1]));
    return 0;
}