zoj-3734-LIKE vs CANDLE

#include<cstdio>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<math.h>
#include<vector>
using namespace std;

int dp[50005][2];
vector<int> p[50005];
int f[50005];
int zhi[50005];
int x,y,n;
int fu,flag;

void dfs(int v)
{
		dp[v][0]=zhi[v];
		dp[v][1]=-zhi[v];

		int i;
		for(i=0;i<p[v].size();i++)
		{
			int u=p[v][i];
			dfs(u);
			dp[v][0]+=dp[u][0];
			dp[v][1]+=dp[u][1];
		}
		if(v!=0){
			if(f[v]) swap(dp[v][0],dp[v][1]);
			dp[v][0]=max(dp[v][0],dp[v][1]-(f[v]?y:x));
			dp[v][1]=max(dp[v][1],dp[v][0]-(f[v]?y:x));
		}
}
int main()
{
    int i,j,k;
	
    while(cin>>n>>x>>y)
	{
		memset(dp,0,sizeof(dp));
		zhi[0]=0;
		for(i=0;i<=n;i++)
			if(!p[i].empty())
				p[i].clear();
		for(i=1;i<=n;i++)
		{
			cin>>zhi[i]>>fu>>f[i]>>flag;
			if(flag==1)
				zhi[i]=-zhi[i];
			p[fu].push_back(i);
		}
		dfs(0);
		if(dp[0][0]<0) printf("HAHAHAOMG\n");
		else
			printf("%d\n",dp[0][0]);
	}
}