看大神的思路很清晰 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;
}