hdu 4399 树形dp

//hdu 4799 树形dp
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <vector>
#define inf 0x3f3f3f3f
#define lowbit(x) ((x)&(-x))
#define maxn 50055
using namespace std;
int dp[maxn][5],val[maxn],sta[maxn],fl[maxn],n,x,y,fa[maxn];
vector<int> g[maxn];

int cal(int a)
{
if(fl[a]==0)
return x;
else
return y;
}

void dfs(int a,int b)
{
int cnt=0;
if(b%2!=0)
sta[a]=1-sta[a];
if(fl[a]==1)
cnt=y;
else
cnt=x;
dp[a][0]=val[a]*(1-sta[a])-sta[a]*val[a];
dp[a][1]=val[a]*sta[a]-(1-sta[a])*val[a]-cnt;
for(int i=0;i<g[a].size();i++)
{
int j=g[a][i];
dfs(j,b+fl[j]);
if(a==0)
dp[a][0]+=max(dp[j][1],dp[j][0]);
else
{
dp[a][1]+=max(dp[j][1]+cal(j),dp[j][0]-cal(j));
dp[a][0]+=max(dp[j][1],dp[j][0]);
}
//printf("dp[%d][1]=%d  dp[%d][0]=%d \n",a,dp[a][1],a,dp[a][0]);
}
}

int main () {
//freopen("d:\\in.txt","r",stdin);
while(scanf("%d%d%d",&n,&x,&y)==3)
{
for(int i=0;i<=n;i++)
g[i].clear();
for(int i=1;i<=n;i++)
{
scanf("%d%d%d%d",&val[i],&fa[i],&fl[i],&sta[i]);
g[fa[i]].push_back(i);
}
val[0]=0;
fl[0]=0;
sta[0]=0;
memset(dp,0,sizeof(0));
dfs(0,0);
if(max(dp[0][0],dp[0][1])<0)
printf("HAHAHAOMG\n");
else
printf("%d\n",max(dp[0][0],dp[0][1]));
}
return 0;
}