Luogu P2146 軟件包管理器(樹鏈剖分+線段樹)
阿新 • • 發佈:2018-10-21
層次 () 必須 rom all const code amp clu
題意
給定\(n\)個軟件包,每個軟件包都有一個依賴軟件包,安裝一個軟件包必須安裝他的依賴軟件包,卸載一個軟件包必須先卸載所有依賴於它的軟件包。給定\(m\)此操作,每次一個操作\(install/unistall\)表示安裝或者卸載。
題解
可以通過簡單畫圖看出,在這個樹形結構的依賴層次圖上,安裝一個包相當於安裝其到根節點路徑上的所有包,刪除一個包相當於刪除其與其子樹的包。用一個重鏈剖分+線段樹處理一下就行了。
#include <cstdio> #include <algorithm> using std::swap; typedef long long ll; const int N = 1e5 + 10; int n, Q, x, ans; int fa[N], dep[N], siz[N], son[N]; int top[N], dfn[N], time; int cnt, from[N], to[N], nxt[N]; int bui[N << 2], set[N << 2]; inline void addEdge(int u, int v) { to[++cnt] = v, nxt[cnt] = from[u], from[u] = cnt; } void dfs1(int u) { siz[u] = 1, dep[u] = dep[fa[u]] + 1; for (int i = from[u]; i; i = nxt[i]) { int v = to[i]; dfs1(v), siz[u] += siz[v]; if(siz[v] > siz[son[u]]) son[u] = v; } } void dfs2(int u, int t) { dfn[u] = ++time, top[u] = t; if(!son[u]) return ; dfs2(son[u], t); for(int i = from[u]; i; i = nxt[i]) { int v = to[i]; if(v == son[u]) continue; dfs2(v, v); } } void modify(int sl, int sr, int k, int o = 1, int l = 1, int r = n) { int len = r - l + 1; if(l >= sl && r <= sr) { if(k == 1) ans += len - bui[o], bui[o] = len; else ans += bui[o], bui[o] = 0; set[o] = k; return ; } int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1; if(set[o]) { if(set[o] == 1) bui[lc] = (len - (len >> 1)), bui[rc] = (len >> 1); else bui[lc] = bui[rc] = 0; set[lc] = set[rc] = set[o], set[o] = 0; } if(sl <= mid) modify(sl, sr, k, lc, l, mid); if(sr > mid) modify(sl, sr, k, rc, mid + 1, r); bui[o] = bui[lc] + bui[rc]; } inline void ins(int x) { int fx = top[x]; while (fx != 1) modify(dfn[fx], dfn[x], 1), x = fa[fx], fx = top[x]; modify(1, dfn[x], 1); } int main () { scanf("%d", &n); for (int i = 2; i <= n; ++i) { scanf("%d", fa + i), ++fa[i]; addEdge(fa[i], i); } dfs1(1), dfs2(1, 1); scanf("%d", &Q); char opt[12]; while(Q--) { scanf("\n%s%d", opt, &x), ans = 0, ++x; if(opt[0] == 'i') ins(x); else modify(dfn[x], dfn[x] + siz[x] - 1, -1); printf("%d\n", ans); } return 0; }
Luogu P2146 軟件包管理器(樹鏈剖分+線段樹)