前言
传送门 :
思路
标签 : 树形 d p dp dp
状态表示 :
f
[
i
]
f[i]
f[i], 以
i
i
i为树根的最大舒适度
状态计算 :
f
[
i
]
=
m
a
x
(
f
[
i
]
,
f
[
j
]
+
w
[
i
]
+
f
[
i
]
)
f[i] =max(f[i],f[j]+w[i]+f[i])
f[i]=max(f[i],f[j]+w[i]+f[i])
记得开 l l ll ll
Mycode
void dfs(int u){
if(st[u]) return;
st[u] = 1;
f[u] += a[u];
for(auto x : g[u]){
if(st[x.to]) continue;
dfs(x.to);
f[u] = max(f[u],f[u]+f[x.to]+x.val);
}
}
void solve()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<n;i++){
int a,b,c;cin>>a>>b>>c;
g[a].pb({b,c});
g[b].pb({a,c});
}
for(int i=1;i<=n;i++){
if(!st[i])
dfs(i);
}
for(int i=1;i<=n;i++){
ans = max(ans,f[i]);
}
cout<<ans<<endl;
}