题意:
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) ;
}