acwing 285 没有上司的舞会 2022/05/11

最新推荐文章于 2024-03-03 18:15:29 发布

三粒小金子

最新推荐文章于 2024-03-03 18:15:29 发布

阅读量1.7k

点赞数

分类专栏： acwing 算法基础课（存答案）文章标签： c++ 算法

本文链接：https://blog.csdn.net/weixin_51658930/article/details/124714199

版权

深度优先搜索图论最大路径算法数据结构

关键词由CSDN通过智能技术生成

acwing 算法基础课（存答案）专栏收录该内容

83 篇文章 1 订阅

订阅专栏

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std ;
const int N = 6010 ;
int happy[N] ;
int e[N] , h[N] , ne[N] , idx ;
bool hasfather[N];
int f[N][2];
int n;

void add(int a , int b)
{
    e[idx] = b , ne[idx] = h[a] , h[a] = idx ++ ;
}
void dfs(int u)
{
    f[u][1] = happy[u];
    for(int i = h[u] ; i != -1 ; i = ne[i])
    {
        int j = e[i] ;
        dfs(j) ;
        
        f[u][0] += max(f[j][1] , f[j][0]) ;
        f[u][1] += f[j][0] ;
    }
}
int main()
{
    cin >> n ;
    for(int i = 1 ; i <= n ; i++)
    {
        cin>> happy[i] ;
    }
    memset(h , -1 , sizeof h) ;
    for(int i = 1 ; i < n ; i++)
    {
        int a , b ;
        cin >> a  >> b ;
        hasfather[a] = true ;
        add(b,a);
    }
    int root = 1;
    while(hasfather[root]) root ++ ;
    dfs(root);
    int res = max(f[root][0] , f[root][1]) ;
    cout << res ;
    return 0 ;
}