洛谷P1122 最大子树和 树形DP初步
小明對數學飽有興趣,并且是個勤奮好學的學生,總是在課后留在教室向老師請教一些問題。一天他早晨騎車去上課,路上見到一個老伯正在修剪花花草草,頓時想到了一個有關修剪花卉的問題。于是當日課后,小明就向老師提出了這個問題:
一株奇怪的花卉,上面共連有NN朵花,共有N-1N?1條枝干將花兒連在一起,并且未修剪時每朵花都不是孤立的。每朵花都有一個“美麗指數”,該數越大說明這朵花越漂亮,也有“美麗指數”為負數的,說明這朵花看著都讓人惡心。所謂“修剪”,意為:去掉其中的一條枝條,這樣一株花就成了兩株,扔掉其中一株。經過一系列“修剪“之后,還剩下最后一株花(也可能是一朵)。老師的任務就是:通過一系列“修剪”(也可以什么“修剪”都不進行),使剩下的那株(那朵)花卉上所有花朵的“美麗指數”之和最大。
老師想了一會兒,給出了正解。小明見問題被輕易攻破,相當不爽,于是又拿來問你。
輸入格式
第一行一個整數N(1 ≤ N ≤ 16000)N(1≤N≤16000)。表示原始的那株花卉上共NN朵花。
第二行有NN個整數,第II個整數表示第II朵花的美麗指數。
接下來N-1N?1行每行兩個整數a,ba,b,表示存在一條連接第aa?朵花和第bb朵花的枝條。
輸出格式
一個數,表示一系列“修剪”之后所能得到的“美麗指數”之和的最大值。保證絕對值不超過21474836472147483647。
輸入輸出樣例
輸入 #1復制
7 -1 -1 -1 1 1 1 0 1 4 2 5 3 6 4 7 5 7 6 7輸出 #1復制
3說明/提示
【數據規模與約定】
對于60\%60%的數據,有N≤1000N≤1000;
對于100\%100%的數據,有N≤16000N≤16000。
這個題,比較簡單,就是求一顆權值最大子樹的權值,記憶化搜索,每次保留最大權值即可。、
#include<bits/stdc++.h> using namespace std; const int maxn=1e5+10; vector<int>E[maxn]; int n,w[maxn],ans1=0; void add(int x,int y) {E[x].push_back(y);E[y].push_back(x); } int dp(int ver,int fa) {int ans=w[ver];for(int i=0;i<E[ver].size();i++)if(E[ver][i]!=fa) ans+=max(0,dp(E[ver][i],ver));ans1=max(ans1,ans);return ans; } int main(){scanf("%d",&n);for(int i=1;i<=n;i++) scanf("%d",w[i]);for(int i=1,x,y;i<=n-1;i++){scanf("%d%d",&x,&y);E[x].push_back(y);//無向圖存 E[y].push_back(x);}dp(1,0); printf("%d\n",ans1);return 0; }?
總結
以上是生活随笔為你收集整理的洛谷P1122 最大子树和 树形DP初步的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 图论--树的直径--DFS+树形DP模板
- 下一篇: 洛谷 P1352 没有上司的舞会(树形