看到数据范围和6s的时限,得(cai)出是一道动态点分治。
这道题有一个巧妙的思路:
假设当前补给站为
u
u
,并强制以为根,
v
v
为的一个子节点,
sumdu
s
u
m
d
u
和
sumdv
s
u
m
d
v
分别为
u
u
的子树内的之和以及
v
v
的子树内的之和,
len(u,v)
l
e
n
(
u
,
v
)
为边
(u,v)
(
u
,
v
)
的长度。
如果将补给站迁移到点
v
v
,那么的子树内的点到补给站的距离减少了
len(u,v)
l
e
n
(
u
,
v
)
,其他的点到补给站的距离增加了
len(u,v)
l
e
n
(
u
,
v
)
。也就是说,补给站迁移到点
v
v
时,代价的增量为:
整理一下,得出性质: u u 为根,为 u u 的子节点,补给站在比 u u 优,当且仅当:
显然满足条件的 v v 最多只有一个。
这时候,如果没有满足条件的,则 u u 为最优位置。否则最优位置在的子树内。
考虑动态点分治时每个点维护 d d ,, sumpd s u m p d 三个值,分别表示:
du d u :点 u u 的值。
sumdu s u m d u :点 u u 的子树内所有的值之和。
sumpdu s u m p d u :如果 u u 为根,则该值等于树中所有点的 dis(v,u)×dv d i s ( v , u ) × d v 之和,否则等于 u u 的子树内所有点的 dis(v,fau)×dv d i s ( v , f a u ) × d v 之和。
( dis(a,b) d i s ( a , b ) 为 a a 和两点在原树中的距离, fau f a u 为点 u u 在分治树上的父节点)
修改比较简单,从开始不断地往 fau f a u 更新三个值即可。
询问时,先把最优解暂定为分治树的根节点。
假设现在到了点 u u 的分治子树,并且发现最优解在的分治子树内( v v 为在分治树上的一个子节点,原树上有边 (u,w) ( u , w ) 并且 w w 在的分治子树内)
则可以转移到 v v 的分治子树内求解。
首先计算出的分治子树内(除 v v 的分治子树内的点之外)的所有点到的 dis×d d i s × d 之和(可以使用 sumpd s u m p d 计算得出),这样 v v 的分治子树之外节点的影响就计算好了。
此外,还要考虑节点的影响因素:实际上对于 u u 的分治子树内但在的分治子树之外的一个点 k k ,上面的计算只是计算好了这一段路径的影响,而 v−>w v − > w 这一段路径的影响还没就算出来。所以,还需要把 dw d w 加上 sumdu−sumdv s u m d u − s u m d v (对应的 sumd s u m d 和 sumpd s u m p d 也要跟着调整),这样就处理好了 v−>w v − > w 这一段路径的影响。这样就能进入 v v 的分治子树进行求解了。但的分治子树求解完毕后,还需要把 dw d w 恢复原值。
最好要使用RMQ求LCA来查询两点距离。
复杂度 O(nlog2n) O ( n log 2 n ) (设 n,q n , q 同阶)
代码:
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define For(i, a, b) for (i = a; i <= b; i++)
#define Edge(u) for (int e = adj[u], v; e; e = nxt[e])
#define Edge2(u) for (int e = adj2[u], v; e; e = nxt2[e])
using namespace std;
inline int read() {
int res = 0; bool bo = 0; char c;
while (((c = getchar()) < '0' || c > '9') && c != '-');
if (c == '-') bo = 1; else res = c - 48;
while ((c = getchar()) >= '0' && c <= '9')
res = (res << 3) + (res << 1) + (c - 48);
return bo ? ~res + 1 : res;
}
typedef long long ll;
const int N = 1e5 + 5, M = N << 1, LogN = 22;
int n, q, ecnt, nxt[M], adj[N], go[M], val[M], sze[N], maxs[N], G,
dep[N], dis[N], m, a[M], Log[M], RMQ[M][LogN], fir[N], fa[N],
Root, ecnt2, nxt2[N], adj2[N], go2[N], sc2[N];
ll d[N], sumd[N], sumpd[N];
bool vis[N];
void add_edge(int u, int v, int w) {
nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v; val[ecnt] = w;
nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u; val[ecnt] = w;
}
void add_edge2(int u, int v, int w) {
nxt2[++ecnt2] = adj2[u]; adj2[u] = ecnt2; go2[ecnt2] = v; sc2[ecnt2] = w;
}
void dfs(int u, int fu) {
dep[u] = dep[fu] + 1; a[fir[u] = ++m] = u;
Edge(u) {
if ((v = go[e]) == fu) continue;
dis[v] = dis[u] + val[e]; dfs(v, u); a[++m] = u;
}
}
void initRMQ()
{
int i, j; For (i, 1, m) RMQ[i][0] = a[i];
For (j, 1, 18) For (i, 1, m - (1 << j) + 1) {
int x = RMQ[i][j - 1], y = RMQ[i + (1 << j - 1)][j - 1];
RMQ[i][j] = dep[x] < dep[y] ? x : y;
}
}
int dist(int u, int v) {
int l = fir[u], r = fir[v]; if (l > r) swap(l, r);
int x = Log[r - l + 1], r1 = RMQ[l][x], r2 = RMQ[r - (1 << x) + 1][x];
int lca = dep[r1] < dep[r2] ? r1 : r2;
return dis[u] + dis[v] - (dis[lca] << 1);
}
void dfs1(int u, int fu) {
sze[u] = 1; for (int e = adj[u], v; e; e = nxt[e]) {
if ((v = go[e]) == fu || vis[v]) continue;
dfs1(v, u); sze[u] += sze[v];
}
}
void dfs2(int r, int u, int fu) {
maxs[u] = sze[r] - sze[u]; Edge(u) {
if ((v = go[e]) == fu || vis[v]) continue;
maxs[u] = max(maxs[u], sze[v]);
}
if (maxs[u] < maxs[G]) G = u;
Edge(u) if ((v = go[e]) != fu && !vis[v]) dfs2(r, v, u);
}
void calcG(int u) {dfs1(u, 0); G = u; dfs2(u, u, 0);}
int dfs3(int u, int fu) {
calcG(u); if (!Root) Root = G; vis[G] = 1; fa[G] = fu; int t = G;
Edge(G) {
if (vis[v = go[e]]) continue;
int w = dfs3(v, t); add_edge2(t, w, v);
}
return t;
}
void change(int u, int delta) {
int v = u; d[u] += delta; while (u) {
sumd[u] += delta;
sumpd[u] += 1ll * delta * dist(v, fa[u] ? fa[u] : u);
u = fa[u];
}
}
ll exc(int u, int w, int x) {
ll ans = 0, cnt = d[u];
Edge2(u) if ((v = go2[e]) != w) ans += sumpd[v], cnt += sumd[v];
d[x] += cnt; return ans + cnt * dist(u, x);
}
ll query(ll ans, int u) {
Edge2(u) if ((sumd[v = go2[e]] << 1) > sumd[u]) {
ll wr = d[sc2[e]], tmp, delta = exc(u, v, sc2[e]);
tmp = d[sc2[e]] - wr; for (int w = sc2[e]; w != u; w = fa[w])
sumd[w] += tmp, sumpd[w] += tmp * dist(sc2[e], fa[w] ? fa[w] : w);
ll nans = 0; for (int z = adj2[v]; z; z = nxt2[z]) nans += sumpd[go2[z]];
ans = delta + query(nans, v); for (int w = sc2[e]; w != u; w = fa[w])
sumd[w] -= tmp, sumpd[w] -= tmp * dist(sc2[e], fa[w] ? fa[w] : w);
d[sc2[e]] = wr; return ans;
}
return ans;
}
int main() {
int i, x, y, z; n = read(); q = read(); Log[0] = -1;
For (i, 1, n << 1) Log[i] = Log[i >> 1] + 1;
For (i, 1, n - 1)
x = read(), y = read(), z = read(), add_edge(x, y, z);
dfs(1, 0); initRMQ(); dfs3(1, 0); while (q--) {
x = read(); y = read(); change(x, y);
printf("%lld\n", query(sumpd[Root], Root));
}
return 0;
}