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]);
}
}