洛谷最大子树和解题报告

最新推荐文章于 2021-11-01 22:01:13 发布

Rushfinen

最新推荐文章于 2021-11-01 22:01:13 发布

阅读量434

点赞数

本文链接：https://blog.csdn.net/Rushfinen/article/details/53645045

版权

不知为什么我做这道题莫名吃力，最后还是看了题解，大概是搜索没有学好的缘故吧。最近总是在学习一些其他的东西，现在看来还是打好基础为妙。这道题其实不难，开三个数组pre[],next[],last[]储存边与点的关系。不妨从1开始搜索，找每条与它相邻的边，依次算出他们的最大子树和，若大于零便累计起来。最后加上1节点的美丽指数，就是以1为根的最大子树和了，其他的同理。

#include<iostream>
using namespace std;
int next[40000],pre[40000],last[40000],book[20000],a[20000],f[20000],num=0,ans=-2147483648;
int merge(int x,int y)
{
    num++;
    next[num]=y;
    pre[num]=last[x];
    last[x]=num; 
}
int search(int x)
{
    int sum=0,i,y,nex;
    if(f[x])
    return f[x];
    i=last[x];
    while(i)
    {
        nex=next[i];
        if(!book[nex])
        {
            book[nex]=1;
            y=search(nex);
            if(y>0)
            sum+=y;
        }
        i=pre[i];
    }
    f[x]=sum+a[x];
    if(f[x]>ans)
    ans=f[x];
    return f[x]; 
}
int main()
{
    int n,x,y;
    cin>>n;
    for(int i=1;i<=n;i++)
    cin>>a[i];
    for(int i=1;i<=n-1;i++)
    {
        cin>>x>>y;
        merge(x,y);
        merge(y,x);
    }
    book[1]=1;
    search(1);
    cout<<ans;
    return 0;
 }