义乌集训Day 6 T3

题目描述
给定一棵大小为 n的有根树。每个节点有个非负权值，对于每个节点 x，我们找到包含 x 的节点的平均值最大的连通块，定义 f(x) 为这个平均值。请求出$$minf(i)│i\in 1,2,\dots ,nminf(i)│i\in 1,2,\dots ,n。$$

看到这题不会换根dp，即使是想到做法了也好像实现不出来，所以当老师和jxc讲的时候就一下子悟了

突破1：

题目叫我们求的是最大值的最小值，这样子很容易让我们想到二分答案，也是是这个最小值

突破2：

怎么去check()二分出来的答案的正确性呢？因为给定的是一颗有根树，所以我们回想到的是树形dp

突破3：
定义树形dp的状态
$f[i]表示取第i个点，加上子树的最大联通块的值$
(先将点的边权减去二分出来的答案，然后如果发现联通快的值大于0，那么return 1)
我们开始就可以以1为根节点进行树形dp,这样就可以求出1为根时候的情况了
$$f[x]=f[x]+max(f[to],0.0);$$

但是如果出现如下图一样的联通块该怎么办呢？

突破4：
在进行一次dp
$g[i]表示以i新为根最大联通块的权值$
$g[to]=f[to]+max(0.0,g[x]-max(f[to],0.0));$
将这个点的to剪掉自己一边就是有另外一边了

#include<bits/stdc++.h>
#define eps 1e-7
#define maxn 100005
using namespace std;
int n;
double l=1e9,r=-1e9;
double a[maxn],b[maxn];
int head[maxn],tot;
double f[maxn],g[maxn];
struct node{
	int to,next;
}edge[maxn*2];
bool flag;
inline void add(int x,int y )
{
	edge[++tot].to=y;
	edge[tot].next=head[x];
	head[x]=tot;
}
inline void dfs1(int x)
{
	for(int i=head[x];i;i=edge[i].next)
	{
		int to=edge[i].to;
		dfs1(to);
		f[x]=f[x]+max(f[to],0.0);
	}
}
inline void dfs2(int x)
{
	if(g[x]<0.0)flag=1;
	for(int i=head[x];i;i=edge[i].next)
	{
		int to=edge[i].to;
		g[to]=f[to]+max(0.0,g[x]-max(f[to],0.0));
		dfs2(to);
	}
}
inline bool check(double x)
{
	flag=0;
	for(int i=1;i<=n;i++)f[i]=a[i]-x;
	dfs1(1);
	g[1]=f[1];
	dfs2(1);
	return flag;
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		l=min(l,a[i]*1.0);
		r=max(r,a[i]*1.0);
	}
	for(int i=2;i<=n;i++)
	{
		int u;cin>>u;
		add(u,i);
	}
	while(r-l>eps)
	{
		double mid=(l+r)/2.0;
		if(check(mid))r=mid;
		else l=mid;
	}
	printf("%.9lf\n",r);
	return 0;
}