【luogu1352】沒有上司的舞會 - 樹形dp
阿新 • • 發佈:2018-04-30
AS true %d -128 can 表示 code 編號 using
題目描述
某大學有N個職員,編號為1~N。他們之間有從屬關系,也就是說他們的關系就像一棵以校長為根的樹,父結點就是子結點的直接上司。現在有個周年慶宴會,宴會每邀請來一個職員都會增加一定的快樂指數Ri,但是呢,如果某個職員的上司來參加舞會了,那麽這個職員就無論如何也不肯來參加舞會了。所以,請你編程計算,邀請哪些職員可以使快樂指數最大,求最大的快樂指數。
輸入輸出格式
輸入格式:
第一行一個整數N。(1<=N<=6000)
接下來N行,第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)
接下來N-1行,每行輸入一對整數L,K。表示K是L的直接上司。
最後一行輸入0 0
輸出格式:
輸出最大的快樂指數。
思路
$ f_{x,0/1} $表示以x為根的子樹,且x不參加(0)參加(1)舞會的最大快樂值
則 $f_{x,0} = \sum_{y \in son(x)}^{ } max \{ f_{y,0},f_{y,1} \}$
$f_{x,1} = {\sum_{y \in son(x)}^{ }f_{y,0} } \ \ + R_{x}$
答案為$ max{f_{root,0},f_{root,1}} $
#include <bits/stdc++.h> using namespace std; const int maxn = 6000+ 10; int n,dp[maxn][2],r[maxn],root; vector<int> edges[maxn]; bool vis[maxn]; inline void dfs(int now) { dp[now][1] = r[now]; for (size_t i = 0;i < edges[now].size();i++) { int to = edges[now][i]; dfs(to); dp[now][0] += max(dp[to][0],dp[to][1]); dp[now][1] += dp[to][0]; } } int main() { scanf("%d",&n); for (int i = 1;i <= n;i++) scanf("%d",&r[i]); for (int i = 1,x,y;i <= n;i++) { scanf("%d%d",&x,&y); edges[y].push_back(x); vis[x] = true; } for (int i = 1;i <= n;i++) if (!vis[i]) { root = i; break; } dfs(root); printf("%d",max(dp[root][0],dp[root][1])); return 0; }
【luogu1352】沒有上司的舞會 - 樹形dp