hdu1520 Anniversary Party(树形dp入门)

题意

一棵树，每个点有权，不能同时选相邻节点，

问最后的最大权值。

二选一肯定是dp啦。

思路来源

https://blog.csdn.net/sr_19930829/article/details/40537507

代码

#include <iostream>
#include <algorithm> 
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <functional>
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const int MOD=998244353;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int> 
#define si set<int>
#define pii pair<int,int> 
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a)) 
using namespace std; 
const int maxn=6e3+5;
vector<int>son[maxn];
bool vis[maxn];
int n,dp[maxn][2];
void dfs(int root)
{
	vis[root]=1;
	for(int i=0;i<son[root].size();++i)
	{
		int v=son[root][i];
		if(!vis[v])
		{
		dfs(v);
		dp[root][1]+=dp[v][0];
		dp[root][0]+=max(dp[v][0],dp[v][1]);
	        }
	} 
} 
void init(int n)
{
	for(int i=0;i<n;++i)
	son[i].clear(),vis[i]=0;
}
int main()
{ 
   while(~scanf("%d",&n))
   {
   	init(n);
   	for(int i=0;i<n;++i)
   	scanf("%d",&dp[i][1]),dp[i][0]=0;
   	int u,v;
	while(~scanf("%d%d",&u,&v)) 
	{
		if(!u&&!v)break;
		u--,v--;
		son[u].push_back(v);
		son[v].push_back(u);
	}
	dfs(0);
	printf("%d\n",max(dp[0][0],dp[0][1]));
   }
   return 0;
}