S
o
u
r
c
e
:
Source:
Source:牛客网暑期ACM多校训练营(第一场)
P
r
o
b
l
e
m
:
Problem:
Problem:一棵树,两点的路径和
d
(
u
,
v
)
d(u,v)
d(u,v)为相邻边权差的平方和的总和。对于每个点,求
max
v
d
(
u
,
v
)
\max_vd(u,v)
maxvd(u,v)
I
d
e
a
:
Idea:
Idea:
先考虑两遍树形DP。
d
o
w
n
(
i
)
down(i)
down(i)表示该点往下可以得到的最大值(包括了自己连向父亲的边),由下往上更新一遍就可以得到。
u
p
(
i
)
up(i)
up(i)表示该点往上再往下可以得到的最大值,从上往下更新。
u
p
(
u
)
=
m
a
x
(
u
p
(
f
a
)
+
(
e
(
u
)
−
e
(
f
a
)
)
2
,
d
o
w
n
(
u
′
)
+
(
e
(
u
)
−
e
(
u
′
)
)
2
)
其
中
f
a
(
u
)
=
f
a
(
u
′
)
up(u)=max(up(fa)+(e(u)-e(fa))^2, down(u')+(e(u)-e(u'))^2) 其中fa(u)=fa(u')
up(u)=max(up(fa)+(e(u)−e(fa))2,down(u′)+(e(u)−e(u′))2)其中fa(u)=fa(u′)
与父亲相同的点考虑一起斜率优化,由于是用另一个数组更新,所以直接按下标sort后从左往右从右往左各更新一遍,保证每对点都会被考虑更新。
由于是最大值,维护上凸包。而斜率递增,所以只需要维护一个栈即可。
C
o
d
e
:
Code:
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;
}