换根dp题集

题目链接

题意：给你一棵树，按顺序输出以每个顶点为根节点，到任意点的边权和最大，输出这个最大值。

思路：换根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