題目描述

某大學有nn個職員，編號為1\ldots n1…n。

他們之間有從屬關係，也就是說他們的關係就像一棵以校長為根的樹，父結點就是子結點的直接上司。

現在有個週年慶宴會，宴會每邀請來一個職員都會增加一定的快樂指數r_iri​，但是呢，如果某個職員的直接上司來參加舞會了，那麼這個職員就無論如何也不肯來參加舞會了。

所以，請你程式設計計算，邀請哪些職員可以使快樂指數最大，求最大的快樂指數。

輸入格式

輸入的第一行是一個整數nn。

第22到第(n + 1)(n+1)行，每行一個整數，第(i+1)(i+1)行的整數表示ii號職員的快樂指數r_iri​。

第(n + 2)(n+2)到第2n2n行，每行輸入一對整數l, kl,k，代表kk是ll的直接上司。

輸出格式

輸出一行一個整數代表最大的快樂指數。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+100;
int n;
vector<int> g[maxn];
int dp[2][maxn];
//dp[0]表示不選擇該節點的最優解
//dp[1]表示選擇該節點的最優解
int a[maxn];
void dfs (int u,int pre) {
    dp[0][u]=0;
    dp[1][u]=a[u];
    for (int v:g[u]) {
        if (v==pre) continue
 
;
        dfs(v,u);
        dp[0][u]+=max(dp[1][v],dp[0][v]);
        dp[1][u]+=dp[0][v];
    }
} 
int main () {
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",a+i);
    for (int i=1;i<n;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(
 
1,0);
    printf("%d\n",max(dp[0][1],dp[1][1]));
}