cf 1092f 树形dp

题意：

n个结点的无向树。每个点权值为ai，每条边权值为1，dist(i,j)表示第i个结点和第j个结点的距离，v可以是任意一个结点。

求的最大值。

题解：

1.大前提是第1个点为根结点。

2.sum1[i]表示第i个结点及其子树的权值和。sumn[i]表示Σ(dist(i,j) *aj)，j为第i个结点的子树上的点。

3.先以第1个结点为根进行一遍dfs（后序遍历），求出sum1和sumn。

4.现在改变sumn的定义，sumn[v]表示题目中的式子的值。再以第1个结点为根进行一遍dfs（前序遍历）。

状态转移方程为sumn[v] += ((sumn[u] - sumn[v] - sum1[v]) + (sum1[1] - sum1[v]))，v是u的子结点。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<vector>
#define N 200005
using namespace std;
int n ;
vector <int> edge[N] ;
bool vis[N] ;
long long sumn[N] ;
long long sum1[N] ;
void cal_sum1_n(int u) //计算结点u及其子树的val和and结点u及其子树的val*dis和 
{
	int i , v ;
	vis[u] = 1 ;
	sumn[u] = 0 ;
	for(i = 0 ; i < edge[u].size() ; i ++)
	{
		v = edge[u][i] ;
		if(vis[v])
		   continue ;
		cal_sum1_n(v) ;
		sum1[u] += sum1[v] ;
		sumn[u] += (sumn[v] + sum1[v]) ;   
	}
}
void cal_sumn(int u)
{
	int i , v ;
	vis[u] = 1 ;
	for(i = 0 ; i < edge[u].size() ; i ++)
	{
		v = edge[u][i] ;
		if(vis[v])
		   continue ;
		sumn[v] += ((sumn[u] - sumn[v] - sum1[v]) + (sum1[1] - sum1[v])) ;
		cal_sumn(v) ;
	}
}
int main()
{
	int i , j , k ;
	int u , v ;
	long long ans = 0 ;
	scanf("%d" , &n) ;
	for(i = 1 ; i <= n ; i ++)
	    scanf("%lld" , &sum1[i]) ;
	for(i = 1 ; i <= n ; i ++)
		edge[i].clear() ;
	for(i = 1 ; i <= n - 1 ; i ++)
	{
		scanf("%d%d" , &u , &v) ;
		edge[u].push_back(v) ;
		edge[v].push_back(u) ;
	}
	memset(vis , 0 , sizeof(vis)) ;
	cal_sum1_n(1) ; //计算以1为根每个点到根的距离 
    memset(vis , 0 , sizeof(vis)) ;
    cal_sumn(1) ; //计算每个点的sumn 
    for(i = 1 ; i <= n ; i ++)
        ans = max(ans , sumn[i]) ;
	printf("%lld\n" , ans) ;
}