作為人生中的第一道樹形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
輸出格式:
輸出最大的快樂指數。
輸入樣例:
1 32 3
6 47 4
4 53 5
0 0輸出樣例:
我們先來分析一下樹形dp是如何工作的:
為了保證樹形dp取點完整且最優,我們就必須要以葉節點為起點,以根節點為終點(因為葉節點的區域相比根節點的區域更稠密)。
這題有個不嚴謹的地方:上司來了,他的職員就不能來,卻沒說是直接上司還是所有的上司;但是根據個人分析和樣例給出的資料來看,應該是直接上司;如果有越級的上司則該職員也可以來(現實也是這樣,職員又很小概率認識自己上司的上司的上司的上司……(此處省略n個上司,n∈z))。
然後,本題的每個點的值都是1,按照下面的樣例可以組成這樣的一棵樹(0 0 的點忽略):
因為如果選了該員工的上司就無法選擇這個員工,所以需要vis陣列判斷改點能否取(他的上司(father)被取了,則所有的職員(son)就不能取);
但是用vis陣列來判斷的話,是不是如果上司改了,他的職員就得全部改掉,甚至還要用到並查集來判斷哪些點的上司的職員,未免有些太麻煩了(如果你會多叉轉二叉,那就呵呵了;但我相信會此演算法的神犇是不會來看該部落格的o(* ̄︶ ̄*)o)。因此,我們會在dp的基礎上再開一維,專門用來判斷當前這個人去還是不去。!!!注意,vis陣列卻必須要用的,這裡的dp[...][0]或dp[...][1]只代表在狀態轉移時用來判斷取還是不取,相當於乙個預處理;實際上你取完之後是要在把他壓到佇列裡的,也就是這個點時實際已經取的,不是用來動態規劃用的!!!
設:dp[x][0]表示以x為根的子樹,且x不參加舞會的最大快樂值;
dp[x][1]表示以x為根的子樹,且x參加了舞會的最大快樂值。
則dp[x][0] = sigma (y是x的兒子)
最後找到唯一的樹根root:
zxy(i,1哎?v是什麼東西??:,n)// i = 1 to n
if(!v[i])
zxy(i,1,n - 1哦!在讀入的時候就把誰是誰的職員壓到佇列裡了,因為根節點就相當於boss,他是沒有上司的,所以說它是不會在v裡的。)
找出根節點在進行dp操作,就可以求出最大子樹了!
ac**:
#pragma gcc optimize(2)#include#include
#include
#include
#include
#include
#include
#include
#include
using
namespace
std;
typedef
long
long
ll;#define n 23333
#define zxy(i,a,b) for(ll i = a ; i <= b ; i++)
#define yxz(i,a,b) for(ll i = a ; i >= b ; i--)ll f[n][
2ll v[n];
vector
long>son[n];
ll mymax(ll a,ll b)
void
zhoier(ll x)
}int
main()
ll mei,yong;
scanf(
"%lld%lld
",&mei,&yong);
ll root;
zxy(i,
1,n)
if(!v[i])
zhoier(root);
//printf("1");
printf("
%lld
",mymax(f[root][0],f[root][1
]));
return0;
}
題解 沒有上司的舞會
5401 沒有上司的舞會 動態規劃 例題 5401 沒有上司的舞會 我是萌新 剛學oi,請問這道樹上dp怎麼做?太難了不會啊 看了題解才會的。dp i,1 0 表示.算了太難了講不清楚.不會怎麼dfs,只會常數記憶體都更小的bfs,球大神教我 winlere include include incl...
To Heart 題解 沒有上司的舞會
ural大學有n個職員,編號為1 n。他們有從屬關係,也就是說他們的關係就像一棵以校長為根的樹,父結點就是子結點的直接上司。每個職員有乙個快樂指數。現在有個周年慶宴會,要求與會職員的快樂指數最大。但是,沒有職員願和直接上司一起參加宴會。第一行乙個整數n。1 n 6000 接下來n行,第i 1行表示i...
To Heart 題解 沒有上司的舞會
ural大學有n個職員,編號為1 n。他們有從屬關係,也就是說他們的關係就像一棵以校長為根的樹,父結點就是子結點的直接上司。每個職員有乙個快樂指數。現在有個周年慶宴會,要求與會職員的快樂指數最大。但是,沒有職員願和直接上司一起參加宴會。第一行乙個整數n。1 n 6000 接下來n行,第i 1行表示i...