codevs1380 沒有上司的舞會
阿新 • • 發佈:2017-06-22
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。
輸出最大的快樂指數。
樣例輸入 Sample Input7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
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 沒有上司的舞會