蓝桥杯 算法训练 ALGO-4 结点选择（链式前向星，动态规划） c/c++

/*

问题描述

有一棵 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的正整数。
*/
分析：
首先了解下什么是链式前向星
https://www.cnblogs.com/LQ-double/p/5971323.html
动态规划
dp[i][0],代表以i作为跟节点的子树，0代表不选i的最大权和值。
dp[i][1],1表示选i的最大权和值。

#include<iostream>
#include<cstring>
#include<cmath>
#define MAX 100010
using namespace std;
//对边做结构 
struct edges{
	int totertex;	//表示该边的终点 
	int nextedge;	//nextdege代表与这条边起点相同的上一条边的编号，为-1则没有 
}edge[MAX*2];
int head[MAX];	//head[i]表示以i顶点为起点的最后一条边， 
int M;	//边的编号 
int dp[MAX][2]; 
void addedges(int from,int to)
{
	edge[M].totertex=to;
	edge[M].nextedge=head[from];
	head[from]=M++;		//head[from]表示以from为起点的最后一条边的编号 
//无向图 
	edge[M].totertex=from;
	edge[M].nextedge=head[to];
	head[to]=M++;
}
void dfs(int x,int pre)
{
	int t;
	for(int i=head[x];i!=-1;i=edge[i].nextedge)
	{
		t=edge[i].totertex;
		if(t==pre)
		continue;
		dfs(t,x);
		dp[x][1]+=dp[t][0];
		dp[x][0]+=max(dp[t][1],dp[t][0]);
	}
	 
}
int main()
{
	int n,a,b;
	cin>>n;
	memset(head,-1,sizeof(head));
	for(int i=1;i<=n;i++)
	cin>>dp[i][1];
	for(int i=0;i<n-1;i++)
	{
		cin>>a>>b;
		addedges(a,b);
	}
	dfs(1,0);
	cout<<max(dp[1][0],dp[1][1])<<endl;
	return 0; 
}