牛客网暑期ACM多校训练营（第二场）H travel

看大神的思路很清晰 https://blog.csdn.net/WuBaizhe/article/details/81153513

重点就是这个 sum[k][i][j]sum[k][i][j]：前kk个儿子所构成的子树林包含jj条不相交路径和ii条以根为端点的不相交链的最大值 
则根据以第kk个儿子为根的子树贡献路径的情况，sum[k]sum[k]的值可以由sum[k−1]sum[k−1]转移而来，故该数组可以使用滚动数组。（路径由链而来）。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
typedef long long  ll;
const int maxn = 400000;
int n,a[maxn+5];
ll d[maxn+2][4],g[maxn+2][4];
vector<int>G[maxn+5];
void Adge(int u,int v)
{
    G[u].push_back(v);
    G[v].push_back(u);
}
void dfs(int u,int fa)
{
    int len = G[u].size(), fr=0;
    ll sum[3][4][4] = {0}, tmp[3][4] = {0};
    for(int i=0; i<len; i++)
    {
        int v = G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        memset(sum[fr],0,sizeof(sum[fr]));
        for(int i=0; i<=3; i++)
        {
            tmp[0][i] = d[v][i];
            tmp[1][i] = g[v][i];
        }
        for(int x=0; x<=2; x++)
            for(int y=0; y<=1&&x+y<=2; y++)
                for(int a=0; a<=3; a++)
                    for(int b=0; a+b<=3; b++)
                    {
                        sum[fr][x+y][a+b] = max(sum[fr][x+y][a+b],sum[fr^1][x][a]+tmp[y][b]);
                    }
        fr^=1;
    }
    for(int i=0; i<=3; i++)
    {
        g[u][i] = max(sum[fr^1][0][i]+a[u],sum[fr^1][1][i]+a[u]);// 肯定要加u这个点，然后这个链可以就是u或者与他儿子链相连
    }
    for(int i=1; i<=3; i++)
    {
        for(int j=0; j<=2; j++)
        {
            d[u][i] = max(d[u][i],sum[fr^1][j][i-1]+a[u]);// 加一条由链而来的路径
        }
    }
    for(int i=0; i<=3; i++) d[u][i] = max(d[u][i],sum[fr^1][0][i]);// 不是由链而来的路径
}
int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]), G[i].clear();
    for(int i=1; i<n; i++)
    {
        int u,v;
        scanf("%d %d",&u,&v);
        Adge(u,v);
    }
    dfs(1,0);
    printf("%lld\n",d[1][3]);
    return 0;
}