树上最长链 (树形dp 或 两次dfs )

带权树上最长链 (可能负权 )
最长链:(树上直径,树上两点距离最大值,树上距离最远的两点 )。

最长链 = 在某个结点的子树中以它为端点的最长链+次长链。
( 次长链可能不存在)

mx1: 以 i 节点为根的子树中,i 节点到叶子节点的最长距离(不含 i 节点)即最长链
mx2: 以 i 节点为根的子树中,i 节点到叶子节点的次长距离(不含 i 节点)即次长链

#define first f
#define second s
#define ll long long
#define pb push_back
#define pii pair<int,char>
#define sl(p) strlen(p)
#define SZ(p) p.size()
#include <bits/stdc++.h>
#define lb lower_bound
#define ub upper_bound
#define mem(a,b) memset(a,b,sizeof(a))
#define fast_cin  ios_base::sync_with_stdio(false),cin.tie(NULL),cout.tie(NULL);
using namespace std;
const ll MOD=1e9+7;
const int maxn=1e6+5;
const int inf=INT_MAX;

struct TMD{
    int to,next;
}edge[maxn<<1];
int head[maxn],cnt;
int n,m,k,q[maxn],p[maxn];
ll dp[maxn],ans=-1e18;
void add(int from,int to)
{
    edge[++cnt]={to,head[from]};
    head[from]=cnt;
}
void dfs(int now,int pre)
{
    ll mx1=0,mx2=0;
    for(int i=head[now];~i;i=edge[i].next){
        int to=edge[i].to;
        if(to==pre) continue;
        dfs(to,now);
        if(dp[to]>mx1) mx2=mx1,mx1=dp[to];
        else if(dp[to]>mx2) mx2=dp[to];
    }
    ans=max(ans,mx1+mx2+p[now]);
    dp[now]=mx1+p[now];
}
int main()
{
    mem(head,-1);
    int u,v,opt;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&p[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(1,0);
    printf("%lld\n",ans);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值