题意:给你一棵树,按顺序输出以每个顶点为根节点,到任意点的边权和最大,输出这个最大值。
思路:换根dp,两遍dfs,第一个dfs求出以这个点为根节点,子树的最大边权值,保存在d数组中。第二个dfs1,首先算出以u节点为根节点的dp值,保存在ans数组里,然后求出u节点的儿子的前缀dp值保存在suf数组中,再求出u节点儿子的后缀dp值保存在x中,那么换根后的u节点的dp值,d【u】=max(suf【u】【i】,x)。
#include<bits/stdc++.h>
#define pi pair<int,int>
using namespace std;
#define mk make_pair
#define ll long long
const int maxn=10010;
vector<pi>G[maxn];
vector<long long>suf[maxn];
ll ans[maxn];
ll d[maxn];
void dfs(int u,int fa)
{
d[u]=0;
for(auto it:G[u])
{
int v=it.first;
int w=it.second;
if(v==fa)continue;
dfs(v,u);
d[u]=max(d[u],d[v]+w);
}
}
void dfs1(int u,int fa)
{
suf[u].push_back(0);
d[u]=0;
for(auto it:G[u])
{
int v=it.first;
int w=it.second;
d[u]=max(d[u],d[v]+w);
suf[u].push_back(d[u]);
}
ans[u]=d[u];
long long x=0;
for(int i=G[u].size()-1;i>=0;i--)
{
int v=G[u][i].first;
int w=G[u][i].secon