【树上斜率优化】18牛客多校1H Longest Path

$Source:$牛客网暑期ACM多校训练营（第一场）
$Problem:$一棵树，两点的路径和$d(u,v)$为相邻边权差的平方和的总和。对于每个点，求${\max }_{v}d(u,v)$
$Idea:$
先考虑两遍树形DP。
$down(i)$表示该点往下可以得到的最大值（包括了自己连向父亲的边），由下往上更新一遍就可以得到。
$up(i)$表示该点往上再往下可以得到的最大值，从上往下更新。
$$up(u)=max(up(fa)+(e(u)-e(fa){)}^2,down(u^{\prime })+(e(u)-e(u^{\prime }){)}^2)其中fa(u)=fa(u^{\prime })$$
与父亲相同的点考虑一起斜率优化，由于是用另一个数组更新，所以直接按下标sort后从左往右从右往左各更新一遍，保证每对点都会被考虑更新。
由于是最大值，维护上凸包。而斜率递增，所以只需要维护一个栈即可。
$Code:$

#include<bits/stdc++.h>
using namespace std;

#define lc o<<1
#define rc o<<1|1
#define fi first
#define se second
#define pb push_back
#define ALL(X) (X).begin(), (X).end()
#define bcnt(X) __builtin_popcountll(X)
#define CLR(A, X) memset(A, X, sizeof(A))
#define DEBUG printf("Passing [%s] in Line %d\n",__FUNCTION__,__LINE__)

using DB = double;
using LL = long long;
using PII = pair<int, int>;
const int N = 1e5+10;
const LL MOD = 1e9+7;
const LL INF = 1e18;
//const DB eps = 1e-8;

int p[N], Q[N];
LL c[N], down[N], up[N];
vector<PII> G[N];

inline LL sqr(const LL &x)  { return x*x; }
inline LL x(int &u) { return c[u]; }
inline LL dx(int &u, int &v) { return x(u)-x(v); }
inline LL y(int &u) { return down[u]+sqr(c[u]); }
inline LL dy(int &u, int &v) { return y(u)-y(v); }

void update(int n) {
    int l = 1, r = 0;
    for(int i = 1; i <= n; i++) {
        int u = p[i];
        while(l<r && dy(Q[r], Q[r-1])<2*c[u]*dx(Q[r], Q[r-1])) r--;
        int v = Q[r];
        if(v) up[u] = max(up[u], down[v]+sqr(c[u]-c[v]));
        while(l<r && dy(Q[r], Q[r-1])*dx(u, Q[r])<dy(u, Q[r])*dx(Q[r], Q[r-1])) r--;
        Q[++r] = u;
    }
    l = 1, r = 0;
    for(int i = n; i >= 1; i--) {
        int u = p[i];
        while(l<r && dy(Q[r], Q[r-1])<2*c[u]*dx(Q[r], Q[r-1])) r--;
        int v = Q[r];
        if(v) up[u] = max(up[u], down[v]+sqr(c[u]-c[v]));
        while(l<r && dy(Q[r], Q[r-1])*dx(u, Q[r])>dy(u, Q[r])*dx(Q[r], Q[r-1])) r--;
        Q[++r] = u;
    }
}
bool cmp(const int &x, const int &y) { return c[x] < c[y]; }
void dfs2(int u,int fa) {
    int k = 0;
    for(PII x:G[u]) if(x.fi != fa) {
        int v = x.fi;
        p[++k] = v;
        if(u != 1) up[v] = max(up[v], up[u]+sqr(c[v]-c[u]));
    }
    sort(p+1, p+k+1, cmp);
    update(k);
    for(PII x:G[u]) if(x.fi != fa) {
        int v = x.fi;
        up[u] = max(up[u], down[v]);
        dfs2(v, u);
    }
}
void dfs1(int u, int fa) {
    down[u] = up[u] = 0;
    for(PII x:G[u]) if(x.fi != fa) {
        int v = x.fi, w = x.se;
        c[v] = w;
        dfs1(v, u);
        down[u] = max(down[u], down[v]+sqr(c[v]-c[u]));
    }
}

int main() {
    int n;
    while(~scanf("%d", &n)) {
        for(int i = 1; i <= n; i++) {
            G[i].clear();
        }
        for(int i = 1; i < n; i++) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            G[u].pb({v, w}), G[v].pb({u, w});
        }
        dfs1(1, 0);
        dfs2(1, 0);
        for(int i = 1; i <= n; i++) {
            printf("%lld\n", up[i]);
        }
    }
    return 0;
}