http://acm.hdu.edu.cn/showproblem.php?pid=6409
思路:
这题直接暴力找每个点的子节点的最大值、最小值、次大值和次小值,求最大值时只加大于0的,最小值只加小于0 的。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+10;
const int inf=0x3f3f3f3f;
vector<int>e[maxn];
struct node
{
int to,next;
}edge[maxn*2];
int head[maxn],tot;
ll w[maxn];
ll dp[maxn][2];
ll dp1[maxn][2];
void addedge(int u,int v)
{
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void init()
{
memset(head,-1,sizeof(head));
memset(dp,0,sizeof(dp));
memset(dp1,0,sizeof(dp1));
tot=0;
}
void dfs(int u,int pre)
{
dp[u][0]=-inf,dp[u][1]=-inf;
dp1[u][0]=inf,dp1[u][1]=inf;
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].to;
if(v==pre)continue;
if(w[v]>dp[u][0])
{
dp[u][1]=dp[u][0];
dp[u][0]=w[v];
}
else if(w[v]==dp[u][0]||w[v]>dp[u][1])
{
dp[u][1]=w[v];
}
if(w[v]<dp1[u][0])
{
dp1[u][1]=dp1[u][0];
dp1[u][0]=w[v];
}
else if(w[v]==dp1[u][0]||w[v]<dp1[u][1])
{
dp1[u][1]=w[v];
}
//printf("%lld %lld %lld %lld\n",dp[u][0],dp[u][1],dp1[u][0],dp1[u][1]);
dfs(v,u);
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
init();
int u;
for(int i=2;i<=n;i++)
{
scanf("%d",&u);
addedge(u,i);
addedge(i,u);
}
for(int i=1;i<=n;i++)
{
scanf("%lld",&w[i]);
}
dfs(1,-1);
dp[0][0]=w[1];
dp1[0][0]=w[1];
ll ma=0;
ll ans=0;
for(int i=0;i<=n;i++)
{
if(dp[i][0]>0)
{
ans+=dp[i][0];
ma=max(dp[i][1],ma);
}
}
ans+=ma;
ll mi=0;
ll ans1=0;
for(int i=0;i<=n;i++)
{
if(dp1[i][0]<0)
{
ans1+=dp1[i][0];
mi=min(dp1[i][1],mi);
}
}
ans1+=mi;
printf("%lld %lld\n",ans,ans1);
}
return 0;
}