樹形dp入門兩題
阿新 • • 發佈:2018-09-06
有一個 cstring scanf node 勤奮 ios con 描述 一行
題目描述
小明對數學飽有興趣,並且是個勤奮好學的學生,總是在課後留在教室向老師請教一些問題。一天他早晨騎車去上課,路上見到一個老伯正在修剪花花草草,頓時想到了一個有關修剪花卉的問題。於是當日課後,小明就向老師提出了這個問題:
一株奇怪的花卉,上面共連有N N朵花,共有N-1N?1條枝幹將花兒連在一起,並且未修剪時每朵花都不是孤立的。每朵花都有一個“美麗指數”,該數越大說明這朵花越漂亮,也有“美麗指數”為負數的,說明這朵花看著都讓人惡心。所謂“修剪”,意為:去掉其中的一條枝條,這樣一株花就成了兩株,扔掉其中一株。經過一系列“修剪“之後,還剩下最後一株花(也可能是一朵)。老師的任務就是:通過一系列“修剪”(也可以什麽“修剪”都不進行),使剩下的那株(那朵)花卉上所有花朵的“美麗指數”之和最大。
老師想了一會兒,給出了正解。小明見問題被輕易攻破,相當不爽,於是又拿來問你。
輸入輸出格式
輸入格式:
第一行一個整數N(1 ≤ N ≤ 16000)N(1≤N≤16000)。表示原始的那株花卉上共N N朵花。
第二行有N N個整數,第II個整數表示第II朵花的美麗指數。
接下來N-1N?1行每行兩個整數a,ba,b,表示存在一條連接第aa 朵花和第bb朵花的枝條。
輸出格式:
一個數,表示一系列“修剪”之後所能得到的“美麗指數”之和的最大值。保證絕對值不超過21474836472147483647。
輸入輸出樣例
輸入樣例:
-1 -1 -1 1 1 1 0
1 4
2 5
3 6
4 7
5 7
6 7
輸出樣例:
3
#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> using namespace std; struct node{ int u,v,nxt; }e[1000101]; int n,m; int fir[1000101],cnt=0,ans=0; int ww[1000011]={0},a[1000101]; void add(int u,int v){ e[++cnt].nxt=fir[u];e[cnt].u=u;e[cnt].v=v;fir[u]=cnt; } void dfs(int u,int fa){ ww[u]=a[u]; for(int i=fir[u];i;i=e[i].nxt){ int v=e[i].v; if(v==fa)continue; dfs(v,u); ww[u]+=max(ww[v],0); } ans=max(ans,ww[u]); } int main(){ scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%d",&a[i]); for(int i=1;i<n;i++){ int x,y; scanf("%d%d",&x,&y); add(x,y);add(y,x); } dfs(1,0); cout<<ans; return 0; }
咳咳,沒時間了,先將代碼發布如上
有空我就補
樹形dp入門兩題