Luogu 4069 [SDOI2016]遊戲
阿新 • • 發佈:2018-12-10
BZOJ 4515
樹鏈剖分 + 李超線段樹
要求支援區間插入一條線段,然後查詢一個區間內的最小值。可以使用李超線段樹解決,因為要維護一個區間內的最小值,所以每一個結點再維護一個$res$表示這個區間內的最小值。因為本題把問題搬到了樹上,剖一下就可以了。
我們可以把一個點$x$到根的距離$dis_x$記為一個點在二維平面上的橫座標,這樣子可以保證一條重鏈上的點的距離遞增,並且不改變點之間的距離。
考慮到一條樹鏈$(x, y, z)$($z$是$lca(x, y)$)上的數字的覆蓋情況是先從$x$向上走到$z$,然後從$z$向下走走到$y$,向上走的時候直線的斜率是$-k$,而向下走的時候直線的斜率是$k$,所以我們把一條樹鏈拆成兩條線段插入。
現在記錄剩下的我寫錯的地方:
1、鏈剖的$dfs2$寫錯了,一定要注意先向下剖分重兒子。
2、在寫線段樹的時候要注意要求出來的線段交點的座標其實是沒有離散化過的值,需要拿它和中間的座標的原來的值進行比較。
3、在求最小的時候注意上層的結點的區間可能包含了我們要求的區間,而這時候上層結點的$res$並不能拿來直接使用(可能會使答案變小),需要根據儲存的直線的解析式重新計算。
時間複雜度$O(nlog^3n)$,然後只要相信它常數很小就好了。
#include <cstdio> #include <cstring> #includeView Code<algorithm> using namespace std; typedef long long ll; typedef double db; const int N = 1e5 + 5; const ll Val0 = 123456789123456789LL; int n, qn, tot = 0, head[N]; int dfsc = 0, id[N], pos[N], fa[N], top[N], dep[N], siz[N], son[N]; ll dis[N], kk[N << 1], bb[N << 1]; structEdge { int to, nxt; ll val; } e[N << 1]; inline void add(int from, int to, ll val) { e[++tot].to = to; e[tot].val = val; e[tot].nxt = head[from]; head[from] = tot; } template <typename T> inline void read(T &X) { X = 0; char ch = 0; T op = 1; for(; ch > '9'|| ch < '0'; ch = getchar()) if(ch == '-') op = -1; for(; ch >= '0' && ch <= '9'; ch = getchar()) X = (X << 3) + (X << 1) + ch - 48; X *= op; } template <typename T> inline void chkMin(T &x, T y) { if(y < x) x = y; } void dfs1(int x, int fat, int depth, ll nowDis) { fa[x] = fat, dep[x] = depth, siz[x] = 1, dis[x] = nowDis; int maxson = -1; for(int i = head[x]; i; i = e[i].nxt) { int y = e[i].to; if(y == fat) continue; dfs1(y, x, depth + 1, nowDis + e[i].val); siz[x] += siz[y]; if(siz[y] > maxson) maxson = siz[y], son[x] = y; } } void dfs2(int x, int topf) { top[x] = topf, pos[id[x] = ++dfsc] = x; if(!son[x]) return; dfs2(son[x], topf); for(int i = head[x]; i; i = e[i].nxt) { int y = e[i].to; if(y == fa[x] || y == son[x]) continue; dfs2(y, y); } } inline int getLca(int x, int y) { for(; top[x] != top[y]; ) { if(dep[top[x]] < dep[top[y]]) swap(x, y); x = fa[top[x]]; } return dep[x] > dep[y] ? y : x; } /* inline ll getDis(int x, int y) { ll z = getLca(x, y); return dis[x] + dis[y] - 2 * dis[z]; } */ namespace SegT { struct Node { int id; ll res; } s[N << 2]; #define lc p << 1 #define rc p << 1 | 1 #define mid ((l + r) >> 1) #define res(p) s[p].res inline void up(int p) { // res(p) = Val0; chkMin(res(p), res(lc)), chkMin(res(p), res(rc)); } void build(int p, int l, int r) { res(p) = Val0, s[p].id = 0; if(l == r) return; build(lc, l, mid); build(rc, mid + 1, r); } void ins(int p, int l, int r, int x, int y, int nid) { if(x <= l && y >= r) { /* if(!s[p].cov) { s[p].cov = 1; s[p].id = nid; chkMin(res(p), min(l1, r1)); return; } */ ll l1 = kk[nid] * dis[pos[l]] + bb[nid], r1 = kk[nid] * dis[pos[r]] + bb[nid]; ll l2 = kk[s[p].id] * dis[pos[l]] + bb[s[p].id], r2 = kk[s[p].id] * dis[pos[r]] + bb[s[p].id]; if(l1 >= l2 && r1 >= r2) return; if(l1 <= l2 && r1 <= r2) { s[p].id = nid; chkMin(res(p), min(l1, r1)); return; } db loc = 1.0 * (bb[nid] - bb[s[p].id]) / (kk[s[p].id] - kk[nid]); db md = (db) dis[pos[mid]]; if(l1 > l2) { if(loc > md) ins(rc, mid + 1, r, x, y, nid); else ins(lc, l, mid, x, y, s[p].id), s[p].id = nid; } else { if(loc > md) ins(rc, mid + 1, r, x, y, s[p].id), s[p].id = nid; else ins(lc, l, mid, x, y, nid); } chkMin(res(p), min(l1, r1)); up(p); return; } if(x <= mid) ins(lc, l, mid, x, y, nid); if(y > mid) ins(rc, mid + 1, r, x, y, nid); up(p); } ll query(int p, int l, int r, int x, int y) { if(x <= l && y >= r) return res(p); ll res = Val0; if(s[p].id) { ll ld = dis[pos[max(l, x)]], rd = dis[pos[min(y, r)]]; chkMin(res, min(ld * kk[s[p].id], rd * kk[s[p].id]) + bb[s[p].id]); } if(x <= mid) chkMin(res, query(lc, l, mid, x, y)); if(y > mid) chkMin(res, query(rc, mid + 1, r, x, y)); return res; } } using namespace SegT; inline void solve(int x, int y) { ll res = Val0; for(; top[x] != top[y]; ) { if(dep[top[x]] < dep[top[y]]) swap(x, y); chkMin(res, query(1, 1, n, id[top[x]], id[x])); x = fa[top[x]]; } if(dep[x] > dep[y]) swap(x, y); chkMin(res, query(1, 1, n, id[x], id[y])); printf("%lld\n", res); } inline void addSeg(int x, int y, int nid) { for(; top[x] != top[y]; ) { // if(dep[top[x]] < dep[top[y]]) swap(x, y); ins(1, 1, n, id[top[x]], id[x], nid); x = fa[top[x]]; } // if(dep[x] > dep[y]) swap(x, y); ins(1, 1, n, id[y], id[x], nid); } int main() { read(n), read(qn); for(int i = 1; i < n; i++) { int x, y; ll v; read(x), read(y), read(v); add(x, y, v), add(y, x, v); } dfs1(1, 0, 1, 0LL), dfs2(1, 1); /* for(int i = 1; i <= n; i++) printf("%lld ", dis[pos[i]]); printf("\n"); */ build(1, 1, n); bb[0] = Val0, kk[0] = 0LL; for(int op, x, y, cnt = 0; qn--; ) { read(op), read(x), read(y); if(op == 1) { ll k, b; read(k), read(b); int z = getLca(x, y); ++cnt; kk[cnt] = -k, bb[cnt] = b + dis[x] * k; addSeg(x, z, cnt); ++cnt; kk[cnt] = k, bb[cnt] = b + (dis[x] - 2LL * dis[z]) * k; addSeg(y, z, cnt); } else solve(x, y); } return 0; }