F. Maximum White Subtree 树形dp*换根

最新推荐文章于 2023-02-07 12:11:10 发布

原创最新推荐文章于 2023-02-07 12:11:10 发布 · 178 阅读

1 ·

CC 4.0 BY-SA版权

CF 专栏收录该内容

7 篇文章

订阅专栏

本文分享了一段使用C++实现的图论算法代码，主要针对树形结构进行深度优先搜索（DFS）来求解节点的最优路径问题。通过递归调用，计算每个节点的最大得分，并进一步调整以确保整体路径的最优化。代码中详细展示了如何初始化图结构、执行DFS遍历以及最终输出结果的过程。

在这里插入图片描述
大佬博客

#include <bits/stdc++.h>
using namespace std ;
const int N = 2e5 + 10 ;
int ans[N] , a[N] , dp[N] ;
vector<int> v[N] ;
int dfs(int u , int fa)
{
  dp[u] = a[u] ;
  for(auto x : v[u])
   {
     if(x == fa) continue ;
     dfs(x , u) ;
     dp[u] += max(dp[x] , 0) ;
   }
}
int dfs1(int u , int fa , int sum)
{
  ans[u] = dp[u] + sum ;
  for(auto x : v[u])
   {
     if(x == fa) continue ;
     dfs1(x , u , max(0 , ans[u] - max(0 , dp[x]))) ;
   }
}
int main()
{
  int n ;
  cin >> n ;
  for(int i = 1; i <= n ;i ++) cin >> a[i] , a[i] = (a[i] == 0 ? -1 : 1);
  for(int i = 1 , x , y ; i < n ;i ++)
   cin >> x >> y , v[x].push_back(y) , v[y].push_back(x) ;
  dfs(1 , -2) ;
  dfs1(1 , -1 , 0) ;
  for(int i = 1; i <= n ;i ++) cout << ans[i] << " " ;
  puts("") ;
  return 0 ;
}