codeforces-743D-树形dp

最新推荐文章于 2020-02-26 20:55:00 发布

zyz_3_14159

最新推荐文章于 2020-02-26 20:55:00 发布

阅读量329

点赞数

分类专栏：树形dp

本文链接：https://blog.csdn.net/zyz_3_14159/article/details/53670438

版权

树形dp 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目大意：给定一棵树，有两个人分别取一个顶点并且它的所有子树，并且这两个人取得部分不可以有重复，问两人取得权值总和最大是多少；

题目解析：先dfs一次另val[i]表示取第i个顶点和它的子树总和是多少,再dfs一次令dp[i]表示在他所包含的子树中，最大值是多少；第三次dfs更新答案ans，dfs到一个点如果有多个儿子的话，就取出他们之中最大的两个dp加起来更新ans；

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#define inf 2000000010
using namespace std;
typedef long long ll;
const int maxn=200010;
vector<int>vec[maxn];
ll dp[maxn],val[maxn];
void dfs(int root,int fa)
{
	for(int i=0;i<vec[root].size();i++)
	{
		int v=vec[root][i];
		if(fa==v)	continue;
		dfs(v,root);
		val[root]+=val[v];
	}
	dp[root]=val[root];
}
void dfs1(int root,int fa)
{
	for(int i=0;i<vec[root].size();i++)
	{
		int v=vec[root][i];
		if(fa==v)	continue;
		dfs1(v,root);
		dp[root]=max(dp[root],dp[v]);
	}
}
ll ans;
void dfs2(int root,int fa)
{
	ll t1,t2;
	t1=-inf;
	t2=-inf;
	for(int i=0;i<vec[root].size();i++)
	{
		if(fa==vec[root][i])	continue;
		dfs2(vec[root][i],root);
		ll t=dp[vec[root][i]];
		if(t>=t1)
		{
			t2=t1;
			t1=t;
			continue;
		}	
		if(t>t2)
		{
			t2=t;
		}
	}
	if(t1!=-inf&&t2!=-inf)
	ans=max(ans,t1+t2);
}
int main()
{
	ll n;
	memset(dp,0,sizeof(dp));
	ans=-inf;
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
		scanf("%lld",&val[i]);
	for(int i=1;i<=n-1;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		vec[u].push_back(v);
		vec[v].push_back(u);
	}
	dfs(1,-1);
	//cout<<1<<endl;
	dfs1(1,-1);
	//cout<<1<<endl;
	dfs2(1,-1);
	if(ans==-inf)
		printf("Impossible\n");
	else 
		printf("%lld\n",ans);
	return 0;
}