1. 程式人生 > >codevs1380 沒有上司的舞會

codevs1380 沒有上司的舞會

bold 提示 ane panel namespace rip 範圍 輸出 c++

題目描述 Description

Ural大學有N個職員,編號為1~N。他們有從屬關系,也就是說他們的關系就像一棵以校長為根的樹,父結點就是子結點的直接上司。每個職員有一個快樂指數。現在有個周年慶宴會,要求與會職員的快樂指數最大。但是,沒有職員願和直接上司一起與會。

輸入描述 Input Description

第一行一個整數N。(1<=N<=6000)
接下來N行,第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)
接下來N-1行,每行輸入一對整數L,K。表示K是L的直接上司。
最後一行輸入0,0。

輸出描述 Output Description

輸出最大的快樂指數。

樣例輸入 Sample Input

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

樣例輸出 Sample Output

5

數據範圍及提示 Data Size & Hint

各個測試點1s

#include<bits/stdc++.h>
using namespace std;
int n,sum[6010][2],fa[6010];

void dp(int now){
	for(int i=1;i<=n;i++)if(fa[i]==now){
		dp(i);
		sum[now][0]+=max(sum[i][0],sum[i][1]);
		sum[now][1]+=sum[i][0];
	}
}

int main(){
	memset(sum,0,sizeof(sum));
	scanf("%d",&n);
	int u,v,top=1;
	for(int i=1;i<=n;i++)scanf("%d",&sum[i][1]);	
	while(scanf("%d%d",&u,&v)){
		if(u==0 && v==0)break;
		fa[u]=v;
	}
	while(fa[top])top=fa[top];
	dp(top);
	printf("%d\n",max(sum[top][1],sum[top][0]));
	return 0;
}

codevs1380 沒有上司的舞會