生命之树(搜索,树的无根转有根)蓝桥杯省赛

    科技2022-07-15  146

    题目描述

    在X森林里,上帝创建了生命之树。

    他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。 上帝要在这棵树内选出一个非空节点集S,使得对于S中的任意两个点a,b,都存在一个点列 {a, v1, v2, …, vk, b} 使得这个点列中的每个点都是S里面的元素,且序列中相邻两个点间有一条边相连。

    在这个前提下,上帝要使得S中的点所对应的整数的和尽量大。 这个最大的和就是上帝给生命之树的评分。

    经过atm的努力,他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算,他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。

    输入

    第一行一个整数 n 表示这棵树有 n 个节点。 第二行 n 个整数,依次表示每个节点的评分。 接下来 n-1 行,每行 2 个整数 u, v,表示存在一条 u 到 v 的边。由于这是一棵树,所以是不存在环的。

    输出

    输出一行一个数,表示上帝给这棵树的分数。

    样例输入

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

    样例输出

    8

    首先将树确定一个根,然后从根开始向下搜索,确定是否加上其子树的值(当子树和是正数就加,因为有贡献,反之不加)因为是讨论子树,所以必定是在同一集合内

    搜索用到了无根转有根的技巧

    #include<cstdio> #include<iostream> #include<vector> using namespace std; const int maxn = 1e6; long long n,w[maxn],ww[maxn],ans; vector<int> ve[maxn];//邻接表 void dfs(int root,int fa){ ww[root]=w[root];//初始化根 for(int i=0;i<ve[root].size();i++){//找出root所有的儿子 int son = ve[root][i]; if(son!=fa){//这是无根转有根的技巧,防止反向搜索 dfs(son,root); if(ww[son]>0){//说明子树是正值,会有贡献 ww[root]+=ww[son];//加上子树的权值和 } } } if(ww[root]>ans)ans=ww[root];//任意一个节点都可以是最大权值和子树的根 } int main(){ cin>>n; for(int i=1;i<=n;i++){ cin>>w[i]; } for(int i=0;i<n-1;i++){ int x,y; cin>>x>>y; ve[x].push_back(y); ve[y].push_back(x); } //输入处理完毕 dfs(1,0); cout<<ans<<endl; return 0; }
    Processed: 0.014, SQL: 8