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

问题描述

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

输入格式

第一行包含一个整数 n 。

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

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

输出格式

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

样例输入

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

样例输出

12

样例说明

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

数据规模与约定

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

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

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

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

思路：用f[i]表示从子树i中选择结点，且结点i必须被选择的最大值，用g[i]表示从子树i中选择结点，且结点i必须不被选择的最大值。

则f[i]=pow[i]+\sum(g[j])，其中pow[i]表示结点i的权值，j是i的子结点，\sum(g[j])表示所有g[j]的和。

g[i]=\sum(max(f[j], g[j]))，其中j是i的子结点。

最后输出f[p],g[p]中较大的数即为答案，p为树的根节点，这道题里任何一点都可以视为根节点。

#include <iostream>
#include<string.h>
using namespace std;

const int e=100100;
int f[e],g[e],visited[e];
int pow[e];
struct child  
{
	int prv;
	child* next;
};//用邻接表存储
child *a[e];//a[i]链表记录了i结点的所有相邻结点
void tree(int u,int v)
{
	child *p,*q;
    p=new child;
    q=new child;
	p->prv=v;
	p->next=a[u];
	a[u]=p;  //以上，u，v两点相邻，故将v结点插入到a[u]的表头
	q->prv=u;
	q->next=a[v];
	a[v]=q; //将u结点插入到a[v]的表头
	
}
void dfs(int  v)//运行dfs(i)会得到f[i]和g[i]的值
{
	visited[v]=1;//将v标为已访问 
	child *p;
	for(p=a[v];p!=NULL;p=p->next)//遍历v的所有邻接点 
	{
		if(visited[p->prv]==0)//若p->prv未访问
		{
			dfs(p->prv);//先得到f[p->prv]和g[p->prv]的值
			f[v]=f[v]+g[p->prv];
			g[v]=g[v]+max(f[p->prv],g[p->prv]);//再应用公式
		}
	}
	f[v]+=pow[v-1];//记得加上pow的值
}
int main()
{
	int n;
	int u,v;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>pow[i];
	}//输入 
	memset(a,NULL,sizeof(a));
	memset(visited,0,sizeof(visited));
	memset(f,0,sizeof(f));
	memset(g,0,sizeof(g));	//初始化 
    for(int i=0;i<n-1;i++)
    {
		cin>>u>>v;
		tree(u,v);//记录u，v结点的信息 
	}
	dfs(1);//可以将结点1视为根节点，然后深搜 
	cout<<max(f[1],g[1])<<endl;
}