标签:舞会 int head tot edge 树形 now dp
很经典的一道树形dp题目
先放下代码,后面补题解。
#include <bits/stdc++.h>
using namespace std;
const int N = 1e4 + 10;
int v[N], tot, head[N], dp[N][2], cnt[N];
struct Edge{
int v, next;
}edge[N];
void add(int u, int v){
edge[tot].next = head[u];
edge[tot].v = v;
head[u] = tot ++;
}
void dfs(int now){
dp[now][0] = 0;
dp[now][1] = v[now];
for(int i = head[now]; i != -1; i = edge[i].next){
int v = edge[i].v;
dfs(v);
dp[now][0] += max(dp[v][0], dp[v][1]);
dp[now][1] += dp[v][0];
}
}
int main(){
int n;
cin >> n;
memset(head, -1, sizeof head);
for(int i = 1; i <= n; i ++)
cin >> v[i];
for(int i = 1; i <= n - 1; i ++){
int u, v;
cin >> u >> v;
add(v, u);
cnt[u] = 1;
}
int node;
for(int i = 1; i <= n; i ++)
if(cnt[i] == 0)
node = i;
dfs(node);
cout << max(dp[node][0], dp[node][1]) << endl;
return 0;
}
标签:舞会,int,head,tot,edge,树形,now,dp 来源: https://www.cnblogs.com/pureayu/p/14979669.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。