codeforces743D 【DFS】

最新推荐文章于 2022-07-21 15:35:04 发布

原创最新推荐文章于 2022-07-21 15:35:04 发布 · 628 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #codeforces #dfs

codeforces 同时被 2 个专栏收录

131 篇文章

订阅专栏

DFS

52 篇文章

订阅专栏

本文介绍了一种使用DFS解决树形DP问题的方法，旨在寻找一棵树中两棵不相交子树的最大和。通过维护每个节点的最大子树和，并确保子树间不相交，最终得出最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：
给你一棵以1为root的根，然后让你求两棵不相交子树的最大和；
思路：
DFS，主要就是你一定得使两棵子树不相交；
对于一个顶点u，维护以u为根的最大子树和。
①：包含u，即所有的结点和。
②：不包含，在子树内找一个最大子树。
每次对于一个根搞搞就好了，维护一下答案；
这个DFS真的赞。。一开始wa4还是那个没有保证子树不相交，后来RE5 and WA5 DFS的函数int应该写LL...（菜啊！
大家都说是树型DP...在弱弱眼里就是个DFS啊ヽ(*。>Д<)o゜......

//赛后qqq神的一句话总结得真好：子树 sum 和子树内的子树最大 sum

#include<bits/stdc++.h>
using namespace std;
typedef __int64 LL;
const LL INF=-1e18;
const int N=2e5+10;
int n;
LL val[N];

struct asd
{
    int to;
    int next;
};

asd q[2*N];
int head[2*N],tol;

void add(int u,int v)
{
    q[tol].to=v;
    q[tol].next=head[u];
    head[u]=tol++;
}

LL ans;
bool vis[N];
int DFS(int u,LL &W)
{
    vis[u]=1;
    LL sum=val[u];
    LL ans1,ans2;
    ans1=ans2=INF;
    for(int i=head[u];i!=-1;i=q[i].next)
    {
        if(vis[q[i].to])
            continue;
        LL tmp;
        sum+=DFS(q[i].to,tmp);
        if(tmp>ans1)
        {
            ans2=ans1;
            ans1=tmp;
        }
        else if(tmp>ans2)
            ans2=tmp;
    }
    if(ans2!=INF)
        ans=max(ans,ans1+ans2);
    W=max(sum,ans1);
    return sum;
}

int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
        scanf("%I64d",&val[i]);
    int u,v;
    tol=0;
    memset(head,-1,sizeof(head));
    for(int i=1; i<n; i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }

    ans=INF;
    memset(vis,0,sizeof(vis));
    LL tmp;
    DFS(1,tmp);
    if(ans!=INF)
        printf("%I64d\n",ans);
    else
        puts("Impossible");
    return 0;
}