问题描述
有一棵 n 个节点的树,树上每个节点都有一个正整数权值。如果一个点被选择了,那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少?
输入格式
第一行包含一个整数 n 。
接下来的一行包含 n 个正整数,第 i 个正整数代表点 i 的权值。
接下来一共 n-1 行,每行描述树上的一条边。
输出格式
输出一个整数,代表选出的点的权值和的最大值。
样例输入
5
1 2 3 4 5
1 2
1 3
2 4
2 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;
}