POJ - 2342 Anniversary party(树形dp入门)
生活随笔
收集整理的這篇文章主要介紹了
POJ - 2342 Anniversary party(树形dp入门)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題目鏈接:點(diǎn)擊查看
題目大意:每個(gè)人都有一個(gè)快樂值,給定一個(gè)樹狀的從屬關(guān)系,僅當(dāng)上司和下屬都不在的時(shí)候這個(gè)個(gè)人的快樂值才能表現(xiàn)出來,問怎么樣才能讓整體的快樂值達(dá)到最大
題目分析:做線段樹做吐了,來換換思維,入門的樹形dp題一道,樹形dp,簡明扼要的說就是在樹上做的動(dòng)態(tài)規(guī)劃,那么就離不開轉(zhuǎn)移方程了,這個(gè)題的狀態(tài)有兩個(gè),來或者不來,有點(diǎn)像背包問題的選與不選,那么我們就假設(shè)dp[i][j]代表第i個(gè)人的狀態(tài)為j時(shí)的最大快樂值,j分為0或1,0代表不來,1代表來,那么可以將第i個(gè)人視為根節(jié)點(diǎn),然后遞歸到葉節(jié)點(diǎn)再往上逐層更新即可,這個(gè)題的轉(zhuǎn)移方程還算簡單的:
dp[i][0]+=max(dp[j][0],dp[j][1]),j為i的兒子,因?yàn)閕不去,所以j去不去無所謂
dp[i][1]+=dp[j][0],j為i的兒子,因?yàn)閕要去,所以j不能去
這個(gè)題還有個(gè)關(guān)鍵點(diǎn)是找到根節(jié)點(diǎn)然后dfs,我是用的入度的關(guān)系,根節(jié)點(diǎn)的入度一定為0,所以加一個(gè)輔助數(shù)組來記錄一下每條邊的入度就好了
直接上代碼了:
#include<iostream> #include<cstdio> #include<string> #include<cstring> #include<algorithm> #include<stack> #include<queue> #include<map> #include<sstream> #include<cmath> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=6e3+100;int dp[N][2];//dp[i][0]表示當(dāng)i為根節(jié)點(diǎn)時(shí)i位置不放的最大值,dp[i][1]表示當(dāng)i為根節(jié)點(diǎn)時(shí)i位置放的最大值 //所以dp[i][0]+=max(dp[j][0],dp[j][1])j為i的兒子,因?yàn)閕不去,所以j去不去無所謂 //dp[i][1]+=dp[j][0]j為i的兒子,因?yàn)閕要去,所以j不能去vector<int>node[N];int num[N];//用來存儲(chǔ)每個(gè)點(diǎn)的入度,若入度為0,則為根節(jié)點(diǎn)void dfs(int u) {for(int i=0;i<node[u].size();i++){int v=node[u][i];dfs(v);dp[u][0]+=max(dp[v][0],dp[v][1]);dp[u][1]+=dp[v][0];} }int main() { // freopen("input.txt","r",stdin);int n;while(scanf("%d",&n)!=EOF){for(int i=1;i<=n;i++){scanf("%d",&dp[i][1]);node[i].clear();}memset(num,0,sizeof(num));int u,v;while(scanf("%d%d",&v,&u)!=EOF&&v+u){node[u].push_back(v);num[v]++;}int root;for(int i=1;i<=n;i++)if(!num[i]){root=i;break;}dfs(root);cout<<max(dp[root][0],dp[root][1])<<endl;}return 0; }?
總結(jié)
以上是生活随笔為你收集整理的POJ - 2342 Anniversary party(树形dp入门)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: HDU - 1542 Atlantis(
- 下一篇: HDU - 5775 Bubble So