标签:rt idx 树状 int ne dfs --- 节点 dp
没有上司的舞会
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 6010;
int hp[N];
int e[N],h[N],ne[N],idx;
int f[N][2];
bool hf[N];
int n;
//邻接表建树
void add(int a,int b){
e[idx] = b;
ne[idx] = h[a];
h[a] = idx ++;
}
void dfs(int u){
//dp赋值
f[u][1] = hp[u];
//遍历子节点
for(int i = h[u];i != -1;i = ne[i]){
int j = e[i];
//子节点更新
dfs(j);
//选不选父节点
f[u][1] += f[j][0];
f[u][0] += max(f[j][1],f[j][0]);
}
}
int main(){
cin >> n;
for(int i = 1;i <= n;i ++) cin >> hp[i];
memset(h,-1,sizeof h);
for(int i = 1;i < n;i ++){
int a,b;
cin >> a >> b;
//b为a的父亲
add(b,a);
hf[a] = 1;
}
//找根节点
int rt = 1;
while(hf[rt]) rt ++;
//根节点开搜
dfs(rt);
cout << max(f[rt][0],f[rt][1]);
}
标签:rt,idx,树状,int,ne,dfs,---,节点,dp 来源: https://www.cnblogs.com/Xuuxxi/p/15413520.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。