洛谷P1352 没有上司的舞会(树形DP水题)
生活随笔
收集整理的這篇文章主要介紹了
洛谷P1352 没有上司的舞会(树形DP水题)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目描述
某大學有N個職員,編號為1~N。他們之間有從屬關系,也就是說他們的關系就像一棵以校長為根的樹,父結點就是子結點的直接上司。現在有個周年慶宴會,宴會每邀請來一個職員都會增加一定的快樂指數Ri,但是呢,如果某個職員的上司來參加舞會了,那么這個職員就無論如何也不肯來參加舞會了。所以,請你編程計算,邀請哪些職員可以使快樂指數最大,求最大的快樂指數。
輸入輸出格式
輸入格式:
?第一行一個整數N。(1<=N<=6000)
接下來N行,第i+1行表示i號職員的快樂指數Ri。(-128<=Ri<=127)
接下來N-1行,每行輸入一對整數L,K。表示K是L的直接上司。
最后一行輸入0 0
輸出格式
輸出最大的快樂指數
輸入輸出樣例
輸入樣例
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
輸出樣例
5樹形DP的水題。
水點一:每個員工只有一個老板,因此,一個father數組足夠存圖。
水點二:狀態賊少:只有去與不去。
那么,我們就來看一看動態轉移方程。
f[i][1]+=f[兒子][0];
f[i][0]+=max(f[兒子][1],f[兒子][0]);
動態轉移方程很好理解,那個i表示以i為跟的子樹,0和1代表兩種狀態,那個兒子當然就是指i的兒子了,怎么找到程序里可以體現。
如果他去了,兒子肯定不去,如果他不去,兒子可去可不去,找一個最優解。
1 #include<iostream> 2 #include<cstdio> 3 #include<cmath> 4 using namespace std; 5 const int MAXN=6000+50; 6 int n; 7 int boss[MAXN]; 8 int dp[MAXN][2]; 9 void tree_dp(int k) 10 { 11 for(int i=1;i<=n;i++) 12 { 13 if(boss[i]==k) 14 { 15 tree_dp(i); 16 dp[k][0]+=max(dp[i][1],dp[i][0]); 17 dp[k][1]+=dp[i][0]; 18 } 19 } 20 } 21 int main() 22 { 23 scanf("%d",&n); 24 for(int i=1;i<=n;i++)scanf("%d",&dp[i][1]); 25 int root=1;//這個根節點亂賦一個值就可以了,要知道,向上找根節點時,不管從哪里開始,最終都會到根節點 26 for(int i=1;i<=n;i++) 27 { 28 int l,k; 29 scanf("%d%d",&l,&k); 30 boss[l]=k;//只保存他老板 31 } 32 while(boss[root]){root=boss[root];}//如果有老板,就一直向樹的跟靠近,直到沒有老板 33 tree_dp(root); 34 printf("%d\n",max(dp[root][1],dp[root][0]));//大老板去與不去中取最優解 35 return 0; 36 }
?
? ?轉載于:https://www.cnblogs.com/pipixue/p/8945758.html
總結
以上是生活随笔為你收集整理的洛谷P1352 没有上司的舞会(树形DP水题)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 微信WebView关闭后本地cookie
- 下一篇: ubuntu开启root登陆