題目描述

某大學有N個職員，編號為1~N。他們之間有從屬關系，也就是說他們的關系就像一棵以校長為根的樹，父結點就是子結點的直接上司。現在有個周年慶宴會，宴會每邀請來一個職員都會增加一定的快樂指數Ri，但是呢，如果某個職員的上司來參加舞會了，那么這個職員就無論如何也不肯來參加舞會了。所以，請你編程計算，邀請哪些職員可以使快樂指數最大，求最大的快樂指數。

輸入格式

第一行一個整數N。(1<=N<=6000)

接下來N行，第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)

接下來N-1行，每行輸入一對整數L,K。表示K是L的直接上司。

最后一行輸入0 0

輸出格式

輸出最大的快樂指數。

輸入輸出樣例

輸入 #1復制 7 1 1 1 1 1 1 1 1 3 2 3 6 4 7 4 4 5 3 5 0 0 輸出 #1復制 5

解析：

樹形DP，設vector數組son[i]為點i的兒子集合
dp[i][1]表示邀請了i這個人物
dp[i][0]表示不邀請i人
根據題意邀請父親兒子就不能來
狀態轉移方程為：
dp[root][0] += std::max(dp[son[root][i]][0],dp[son[root][i]][1])
dp[root][1] += dp[son[root][i]][0]

1 #include <cstdio> 2 #include <iostream> 3 #include <cmath> 4 #include <cstring> 5 #include <algorithm> 6 #include <queue> 7 #include <stack> 8 #include <vector> 9 #define Max 6050 10 #define re register 11 std::vector<int>son[Max]; 12 int n,root,dp[Max][2],fa[Max]; 13 void dfs(int root) { 14 for(re int i = 0 ; i < son[root].size() ; ++ i) 15 dfs(son[root][i]); 16 for(re int i = 0 ; i < son[root].size() ; ++ i) { 17 dp[root][0] += std::max(dp[son[root][i]][0],dp[son[root][i]][1]); 18 dp[root][1] += dp[son[root][i]][0]; 19 } 20 } 21 void init() { 22 scanf("%d",&n);int u,v;memset(fa,-1,sizeof fa); 23 for(re int i = 1 ; i <= n ; ++ i) scanf("%d",&dp[i][1]); 24 for(re int i = 1 ; i < n ; ++ i) 25 scanf("%d%d",&u,&v),fa[u]=v,son[v].push_back(u); 26 scanf("%d%d",&u,&v); 27 } 28 inline void print(int root) {printf("%d",std::max(dp[root][0],dp[root][1]));} 29 void work() { 30 int root=1; 31 while(fa[root] != -1) root = fa[root]; 32 dfs(root); 33 print(root); 34 } 35 int main() { 36 init(); 37 work(); 38 return 0; 39 } AC 代碼

轉載于:https://www.cnblogs.com/handsomegodzilla/p/11359388.html

總結

以上是生活随笔為你收集整理的洛谷P1352 没有上司的舞会题解的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。