（树形dp）洛谷P1352没有上司的舞会

洛谷P1352没有上司的舞会

思路：

树形dp的模板题。
dp[i][0]表示不i结点后i子树能得到得最大值，dp[i][1]表示选了i子树后能得到得最大值。
用dfs，回溯时通过子节点向上更新信息。
dp[u][0]=∑max{dp[v][1],dp[v][0]};
dp[u][1]=a[u]+dp[v][0];
对于找根节点，我们可以把{u，v}作为一个有向边，根节点即为入度为0的点。

代码：

#include
const int N=1e5+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
int dp[N][2],a[N],in[N];
vector<int> son[N];
void dfs(int u)
{
	dp[u][0]=0;
	dp[u][1]=a[u];
	for(int i=0;i<son[u].size();i++)
	{
		int v=son[u][i];
		dfs(v);
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		dp[u][1]+=dp[v][0];
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,i;
	cin>>n;
	for(i=1;i<=n;i++)
		cin>>a[i];
	for(i=1;i<=n-1;i++)
	{
		int u,v;
		cin>>u>>v;
		son[v].push_back(u);
		in[u]=1;
	}
	int root;
	for(i=1;i<=n;i++)
		if(!in[i])
		{
			root=i;
			break;
		}
	dfs(root);
	int ans=max(dp[root][0],dp[root][1]);
	cout<<ans<<endl;
	return 0;
}