codefrce1187E Tree Painting(DFS)
题目大意
给出一棵树,其所有的节点为白色,现在要将其染为黑色。
一开始首先在树上随意选择一个点将其染成黑色,接下来每次只能选择与黑色的点相联通的点染成黑色。其中每次选择一个点时,其就将产生一个贡献,其贡献的大小为选择的这个点在为白色时,其所在的白色联通块的大小,问如何染色可以使得最终的贡献之和尽可能得大
解题思路
由于图为树,故显然可得权值的大小只与选择的第一个节点有关。
为此只需要算出每个节点的答案,最后取其中最大的那个即可。首先随意地选择一个根,将无根树转化为有根树。接下来从根开始对树进行DFS,算出每个节点的子树之值。接下来再进行一次DFS,从根节点开始每个节点的真正值就是其父节点的值减去父节点的包含该节点的子树的值再加上除了该点子树书剩余节点的数量(即将父节点作为一开始选择的节点转化为将当前点作为初始选择的点)。
最后枚举每个节点的答案选出最大值即可
AC代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef long long LL;
const int size=2e5+5;
vector<int> Edge[size];
int sz[size];LL val[size];
int n;
void down(int u,int p)
{
sz[u]=1;
for(auto v:Edge[u])
{
if(v==p) continue;
down(v,u);
val[u]+=val[v];
sz[u]+=sz[v];
}
val[u]+=sz[u];
}
void up(int u,int p)
{
if(p!=0)
{
val[u]=val[u]+val[p]-sz[u]-val[u]+(n-sz[u]);
}
for(auto v:Edge[u]) if(v!=p)up(v,u);
}
int32_t main()
{
scanf("%lld",&n);
int u,v;
for(int i=1;i<n;i++)
{
scanf("%lld%lld",&u,&v);
Edge[u].push_back(v);
Edge[v].push_back(u);
}
down(1,0);
up(1,0);
LL ans=0;
for(int i=1;i<=n;i++) ans=max(ans,val[i]);
printf("%lld\n",ans);
}