没有上司的舞会

有了上一篇博客,没有看上一篇博客的可以看看上一篇博客,我们对没有上司的舞会这道题会有更好的理解~

所以关键的思路就是确定对于每一个节点我们应该维护什么内容才是最合适的,这个题目和上一篇博客的最后一道题目很相似,我们思考后发现每个节点只有选和不选两种状态,有了这个想法

写起来就很轻松了,其实思考维护什么状态就是要看看我们设置啥样的状态才能计算出要求的值并且还要保证在求的过程中维护好题目要求的规则

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
int e[N],ne[N],h[N],idx;
int n;
int ha[N];
int f1[N][2];
int f[N];

void add(int a,int b){
	e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}

void dfs(int u,int father){
	
	f1[u][0] = 0,f1[u][1] = ha[u];
	for(int i=h[u];~i;i=ne[i]){
		int j = e[i];
		
		if(j==father)continue;
		dfs(j,u);
		
		f1[u][0] = f1[u][0] + max(f1[j][1],f1[j][0]);
		f1[u][1] = f1[u][1] + f1[j][0];
		
	}
	
}


int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)cin>>ha[i];
	memset(h,-1,sizeof h);
	
	for(int i=1;i<n;i++){
		int a,b;cin>>a>>b;
		add(a,b),add(b,a);
		f[a] = b;
	}
	
	int root=1;
	while(f[root])root++;
	
	dfs(root,-1);
	
	cout<<max(f1[root][0],f1[root][1]);
	
}
相关推荐
焦耳加热2 小时前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
wan5555cn2 小时前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
u6063 小时前
常用排序算法核心知识点梳理
算法·排序
蒋星熠5 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油5 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream5 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL5 小时前
【c++】c++第一课:命名空间
数据结构·c++·算法
空白到白6 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技6 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
zzzsde6 小时前
【数据结构】队列
数据结构·算法