Problem Description

度度熊、光羽、帶勁三個人是好朋友。

度度熊有一棵n個點的有根樹，其中1號點為樹根。除根節點之外，每個點都有父節點，記i號點的父節點為fa[i]。

度度熊稱點i和點j是**兄弟**（其中i≠j）當且僅當fa[i]=fa[j]。

第i個點的權值為Ai。現要求選出一個點集，該點集合法當且僅當**點集中至多隻有一對兄弟**。

度度熊想知道，在所有可行的點集中，權值和**最大**以及**最小**的點集權值和分別是多少？

Input

第一行一個數，表示資料組數T。

每組資料第一行一個整數n；第二行n−1個數，表示fa[2],fa[3],..,fa[n]；第三行n個數，表示Ai。

資料組數T=100，滿足：

- 1≤n≤105
- −109≤Ai≤109
- 1≤fa[i]<i

其中90%的資料滿足n≤1000。

Output

每組資料輸出一行，每行包含兩個數，分別表示權值和的**最大值**和**最小值**。

Sample Input

2

5

1 1 2 2

-4 -4 -1 -2 -5

5

1 1 3 2

-1 -4 2 0 -2

Sample Output

0 -15 
2 -7

思路：這道題最容易想到的就是貪心，最大值時只取大於0的權值，最小值只去小於0的權值。等我學會了樹形DP再回來優化..

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
typedef long long ll;
const int N = 1e5;
struct node {
	int w, fa;
};
node a[N];
//對結構體排序
bool cmp1(node x, node y)
{
	if (x.w > y.w)
		return true;
	else
		return false;
}
bool cmp2(node x, node y)
{
	if (x.w < y.w)
		return true;
	else
		return false;
}
void solve() 
{
	int n;
	cin >> n;
	int mp[N] = { 0 };
	a[1].fa = N - 1;
	for (int i = 2; i <= n; i++)
	{
		cin >> a[i].fa;
	}
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i].w;
	}
	sort(a + 1, a + n + 1, cmp1);
	ll sum1 = 0, sum2 = 0;
	int flag1 = 0, flag2 = 0;
	for (int i = 1; i <= n; i++)
	{
		int fa = a[i].fa, w = a[i].w;
		if (mp[fa] == 1 && flag1 == 0 && w>0)//flag用來標誌至多隻有一對兄弟的要求
		{
			flag1 = 1;
			sum1 += w;
		}
		else if (mp[fa] == 0 && w>0)
		{
			mp[fa]++;
			sum1 += w;
		}
		else if (w <= 0)
		{
			break;
		}
	}
	memset(mp, 0, sizeof(mp));
	sort(a + 1, a + n + 1,cmp2);
	for (int i = 1; i <= n; i++)
	{
		int fa = a[i].fa, w = a[i].w;
		if (mp[fa] == 1 && flag2 == 0 && w<0)//至多一個兄弟結點
		{
			flag2 = 1;
			sum2 += w;
		}
		else if (mp[fa] == 0 && w<0)
		{
			mp[fa]++;
			sum2 += w;
		}
		else if (w >= 0)
		{
			break;
		}
	}
	cout << sum1 << " " << sum2 << endl;
}
int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		solve();
	}
	return 0;
}
 

，