洛谷-P1352没有上司的舞会

基本思路

树形dp入门题，0表示不选中1表示选中。

\[\begin{array}{l} treedp[u][0]=max(treedp[u.boss][0],treedp[u.boss][1])\\ treedp[u][1]=treedp[u.boss][0]+arr[u]; \end{array} \]

#include<bits/stdc++.h>
using namespace std;
vector<int>boss[7000];
int arr[7000],brr[7000];
int treedp[7000][2];
int n;
void dfs(int u)
{
	treedp[u][1]=arr[u];
	treedp[u][0]=0;
	//for(int i=1;i<=n;i++)
	{
		for(int v:boss[u])
		{
			dfs(v);
			treedp[u][0]+=max(treedp[v][0],treedp[v][1]);
			treedp[u][1]+=treedp[v][0];
		}
	}
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>arr[i];
		
	}
	
	for(int i=1;i<=n-1;i++)
	{
		int l,r;
		cin>>l>>r;
		boss[r].push_back(l);
		brr[l]=1;
	}
	for(int i=1;i<=n;i++)
	{
		if(brr[i]==0)
		{
			//cout<<i<<endl;
			dfs(i);
			cout<<max(treedp[i][0],treedp[i][1])<<endl;
			break;
		}
	}
	
	return 0;
}