蓝桥杯 ALGO-4 算法训练结点选择

问题描述

有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？

输入格式

第一行包含一个整数 n 。

接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。

接下来一共 n-1 行，每行描述树上的一条边。

输出格式

输出一个整数，代表选出的点的权值和的最大值。

样例输入

5
1 2 3 4 5
1 2
1 3
2 4
2 5

样例输出

样例说明

选择3、4、5号点，权值和为 3+4+5 = 12 。

数据规模与约定

对于20%的数据， n <= 20。

对于50%的数据， n <= 1000。

对于100%的数据， n <= 100000。

权值均为不超过1000的正整数。

分析：题目给出的数据不一定是二叉树，所以可以看作图来处理～其实就是用邻接表存储啦～v[i]数组中保存i结点的孩子节点们～dp[i][0]表示不取i结点的结果～dp[i][1]表示取i结点的结果～

用深度优先搜索+动态规划，每个点的最大权值有取当前这个点和不取当前这个点两种情况～如果取当前点，则不能取与它相邻的任何点；不取当前点，则取与它相邻点的最大值进行累加～从底向上累加到顶部～max(dp[1][0], dp[1][1])就是所求结果～

用一个变量pre保存当前结点的前一个结点～如果等于pre说明访问到了它的父亲结点，为了防止重复访问，要在v[node][i]不等于pre时候继续dfs下去～否则可能会形成无限循环的环～

#include <iostream>
#include <vector>
using namespace std;
int dp[100010][2];
vector<vector<int> > v;
void dfs(int node, int pre) {for (int i = 0; i < v[node].size(); i++) {int temp = v[node][i];if (temp != pre) {dfs(temp, node);dp[node][0] += max(dp[temp][0], dp[temp][1]);dp[node][1] += dp[temp][0];}}
}
int main() {int n, a, b;scanf("%d", &n);for (int i = 1; i <= n; i++)scanf("%d", &dp[i][1]);v.resize(n + 1);for (int i = 1; i <= n - 1; i++) {scanf("%d%d", &a, &b);v[a].push_back(b);v[b].push_back(a);}dfs(1, 0);cout << max(dp[1][0], dp[1][1]);return 0;
}

蓝桥杯 ALGO-4 算法训练结点选择相关推荐

蓝桥杯：试题算法训练 Remember the A La Mode
蓝桥杯:试题算法训练 Remember the A La Mode 资源限制时间限制:1.0s 内存限制:256.0MB 问题描述 Hugh Samston经营着一个为今年的ICPC世界总决赛的参 ...
矩阵乘法算法训练试题_蓝桥杯习题集_ 算法训练矩阵乘法
原文: 算法训练矩阵乘法时间限制:1.0s 内存限制:512.0MB 问题描述输入两个矩阵,分别是m*s,s*n大小.输出两个矩阵相乘的结果. 输入格式第一行,空格隔开的三个正整数m,s, ...
蓝桥杯算法训练结点选择
问题描述有一棵 n 个节点的树,树上每个节点都有一个正整数权值.如果一个点被选择了,那么在树上和它相邻的点都不能被选择.求选出的点的权值和最大是多少? 输入格式第一行包含一个整数 n . 接下来的 ...
蓝桥杯：试题算法训练采油区域矩阵前缀和+动态规划+分治+枚举
资源限制时间限制:2.0s 内存限制:512.0MB 采油区域 Siruseri政府决定将石油资源丰富的Navalur省的土地拍卖给私人承包商以建立油井.被拍卖的整块土地为一个矩形区域,被划分为M× ...
蓝桥杯练习系统算法训练习题加答案java版本
附上百度文库的链接:http://wenku.baidu.com/view/afb78d36b42acfc789eb172ded630b1c59ee9bf7 转载于:https://www.cnblo ...
java蓝桥杯的题_Java蓝桥杯试题集——算法训练ALGO-116——最大的算式
题目要求解题思路动态规划,今天才弄明白QAQ,借鉴了这位大佬的博客,曹磊的博客写的很好!但是我觉得我的循环方式更容易理解嘿嘿嘿~ 首先建立如下图的数组,行数代表前几位数,列数代表有几个乘号.将第 ...
蓝桥杯练习系统—算法训练 s01串
第一部分:题目问题描述 s01串初始为"0" 按以下方式变换 0变1,1变01 输入格式 1个整数(0~19) 输出格式 n次变换后s01串样例输入 3 样例输出 101 数据 ...
蓝桥杯-黑色星期五（算法训练）
题目描述: 有些西方人比较迷信,如果某个月的13号正好是星期五,他们就会觉得不太吉利,用古人的说法,就是"诸事不宜".请你编写一个程序,统计出在某个特定的年份中,出现了多少次既是1 ...
【算法】【蓝桥杯】试题算法训练黑心药商 ALGO-1002
题目资源限制时间限制:1.0s 内存限制:256.0MB 问题描述 JiaoShou消灭了百变怪,为爱琳世界赢得了和平,但他突然发现自己没有升级,这就意味着必须去喝药补血.爱琳世界的NPC卖的药已 ...

蓝桥杯 ALGO-4 算法训练结点选择

蓝桥杯 ALGO-4 算法训练结点选择相关推荐

最新文章

热门文章

蓝桥杯 ALGO-4 算法训练 结点选择

蓝桥杯 ALGO-4 算法训练 结点选择相关推荐

最新文章

热门文章

蓝桥杯 ALGO-4 算法训练结点选择

蓝桥杯 ALGO-4 算法训练结点选择相关推荐