#6073. 「2017 山東一輪集訓(xùn) Day5」距離

給定一顆有$n$個(gè)節(jié)點(diǎn)帶邊權(quán)的樹，以及一個(gè)排列$p$，$path(u,v)$為$u,v$路徑上的點(diǎn)集，$dist(u,v)$為$u,v$之間的最短路的長(zhǎng)度。

有$m$次詢問，每次給定$u,v,k$，要求$\sum _{i\in path(u,v)}dist(p_i,k)$，要求在線求解。

我們選定$1$號(hào)節(jié)點(diǎn)為根節(jié)點(diǎn)，定義$d(i)$為點(diǎn)$i$到根節(jié)點(diǎn)的距離，考慮初始的$\sum _{i\in paht(u,v)}dist(p_i,k)$，有如下：
$$\sum _{i\in path(u,v)}dist(p_i,k)=\sum _{i\in path(u,v)}\left(d(p_i)+d(k)?2\times d(lca(p_i,k))\right)$$
對(duì)于$d(k)$的計(jì)算，由于$d(k)$是一個(gè)定值，則這一部分的答案即為$u,v$間點(diǎn)的個(gè)數(shù)$ \times d(k)$。

對(duì)于$d(p_i)$的計(jì)算，我們考慮樹上差分前綴和來求解，我們定義$sum[n]=\sum _{i\in paht(1,n)}d(p_i)$，

則$\sum _{i\in path(u,v)}d(p_i)=sum[u]+sum[v]?sum[lca(u,v)]?sum[fa(lca(u,v))]$。

最后一步，考慮最難算的$\sum _{i\in path(u,v)}d(lca(p_i,k))$，可以仿照P4211 [LNOI2014]LCA這題的計(jì)算方式，

由于強(qiáng)制在線，所以這題必須用主席樹，我們定義點(diǎn)$i$所代表的主席樹為從$1?>i$上，也就是根節(jié)點(diǎn)到$i$上，

點(diǎn)$p_i?>1$所代表的信息，最后我們只需要四顆主席樹即可解決問題$u+v?lca(u,v)?fa(lca(u,v))$。

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N = 2e5 + 10;int head[N], to[N << 1], nex[N << 1], value[N << 1], cnt = 1;int p[N], n, m, type;int fa[N], sz[N], son[N], top[N], rk[N], id[N], dep[N], w[N], tot;int root[N], ls[N * 100], rs[N * 100], num;ll dis[N], len[N], s[N], sum[N * 100], lazy[N * 100];void add(int x, int y, int w) {to[cnt] = y;nex[cnt] = head[x];value[cnt] = w;head[x] = cnt++; }void dfs1(int rt, int f) {fa[rt] = f, dep[rt] = dep[f] + 1, sz[rt] = 1;for (int i = head[rt]; i; i = nex[i]) {if (to[i] == f) {continue;}dis[to[i]] = dis[rt] + value[i];dfs1(to[i], rt);w[to[i]] = value[i];sz[rt] += sz[to[i]];if (!son[rt] || sz[son[rt]] < sz[to[i]]) {son[rt] = to[i];}} }void dfs2(int rt, int tp) {rk[++tot] = rt, id[rt] = tot, len[tot] = w[rt], top[rt] = tp;if (!son[rt]) {return ;}dfs2(son[rt], tp);for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa[rt] || to[i] == son[rt]) {continue;}dfs2(to[i], to[i]);} }int lca(int u, int v) {while (top[u] != top[v]) {if (dep[top[u]] < dep[top[v]]) {swap(u, v);}u = fa[top[u]];}return dep[u] < dep[v] ? u : v; }void update(int &rt, int pre, int l, int r, int L, int R) {rt = ++num;ls[rt] = ls[pre], rs[rt] = rs[pre], sum[rt] = sum[pre], lazy[rt] = lazy[pre];sum[rt] += len[min(r, R)] - len[max(l, L) - 1];if (l >= L && r <= R) {lazy[rt] += 1;return ;}int mid = l + r >> 1;if (L <= mid) {update(ls[rt], ls[pre], l, mid, L, R);}if (R > mid) {update(rs[rt], rs[pre], mid + 1, r, L, R);} }ll query(int u, int v, int f, int ff, int l, int r, int L, int R) {if (l >= L && r <= R) {return sum[u] + sum[v] - sum[f] - sum[ff];}ll ans = (lazy[u] + lazy[v] - lazy[f] - lazy[ff]) * (len[min(r, R)] - len[max(l, L) - 1]);int mid = l + r >> 1;if (L <= mid) {ans += query(ls[u], ls[v], ls[f], ls[ff], l, mid, L, R);}if (R > mid) {ans += query(rs[u], rs[v], rs[f], rs[ff], mid + 1, r, L, R);}return ans; }void dfs(int rt, int f) {s[rt] = s[f] + dis[p[rt]];int cur = p[rt];root[rt] = root[f];while (cur) {update(root[rt], root[rt], 1, n, id[top[cur]], id[cur]);cur = fa[top[cur]];}for (int i = head[rt]; i; i = nex[i]) {if (to[i] == f) {continue;}dfs(to[i], rt);} }ll query(int u, int v, int f, int ff, int rt) {ll ans = 0;while (rt) {ans += query(root[u], root[v], root[f], root[ff], 1, n, id[top[rt]], id[rt]);rt = fa[top[rt]];}return ans; }int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);scanf("%d %d %d", &type, &n, &m);for (int i = 1, u, v, w; i < n; i++) {scanf("%d %d %d", &u, &v, &w);add(u, v, w);add(v, u, w);}for (int i = 1; i <= n; i++) {scanf("%d", &p[i]);}dfs1(1, 0);dfs2(1, 1);for (int i = 1; i <= n; i++) {len[i] += len[i - 1];}dfs(1, 0);ll ans = 0;for (int i = 1, u, v, k; i <= m; i++) {scanf("%d %d %d", &u, &v, &k);u = u ^ (ans * type), v = v ^ (ans * type), k = k ^ (ans * type);int f = lca(u, v), ff = fa[f];ans = (dep[u] + dep[v] - dep[f] - dep[ff]) * dis[k];ans += s[u] + s[v] - s[f] - s[ff];ans -= 2 * query(u, v, f, ff, k);printf("%lld\n", ans);}return 0; }

總結(jié)

以上是生活随笔為你收集整理的#6073. 「2017 山东一轮集训 Day5」距离（树链剖分 + 永久标记主席树）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。