1. 程式人生 > >bzoj 3779 重組病毒

bzoj 3779 重組病毒

sol:

題面看吐了,實際上操作就是

1.access

2.makeroot

3.查詢一個點到根上有多少虛邊

 

1. 和 2. 可以 access ,每次斷掉的輕邊子樹 -1,新連的輕邊子樹 +1 ,用線段樹維護一下子樹即可,類似樹剖套線段樹的方法換根

順便複習一下換根

1.如果當前點在根的子樹裡,直接加

2.如果當前點就是根,整棵樹加

3.如果根在當前點的子樹裡,從根爬到 x 下面的一個兒子 p ,修改除 p 子樹外所有點

#include<bits/stdc++.h>
#define LL long long
using
namespace std; inline int read() { int x = 0,f = 1;char ch = getchar(); for(;!isdigit(ch);ch = getchar())if(ch == '-') f = -f; for(;isdigit(ch);ch = getchar())x = 10 * x + ch - '0'; return x * f; } const int maxn = 100010; int n,m,rt; int first[maxn],to[maxn << 1],nx[maxn << 1
],cnt; inline void add(int u,int v) { to[++cnt] = v; nx[cnt] = first[u]; first[u] = cnt; } inline void ins(int u,int v){add(u,v);add(v,u);} int par[maxn][24],ind[maxn],oud[maxn],size[maxn],dep[maxn],reh[maxn],dfn; inline void dfs(int x) { size[x] = 1;ind[x] = ++dfn;reh[dfn] = dep[x];
for(int i=1;i<=23;i++)par[x][i] = par[par[x][i - 1]][i - 1]; for(int i=first[x];i;i=nx[i]) { if(to[i] == par[x][0])continue; par[to[i]][0] = x; dep[to[i]] = dep[x] + 1; dfs(to[i]);size[x] += size[to[i]]; }oud[x] = dfn; } LL seg[maxn << 2],tag[maxn << 2]; inline void pushup(int x){seg[x] = seg[x << 1] + seg[x << 1 | 1];} inline void pushdown(int x,int l,int r) { if(tag[x]) { int mid = (l + r) >> 1; tag[x << 1] += tag[x];tag[x << 1 | 1] += tag[x]; seg[x << 1] += (mid - l + 1) * tag[x];seg[x << 1 | 1] += (r - mid) * tag[x]; tag[x] = 0; } } inline void build(int x,int l,int r) { if(l == r) { seg[x] = reh[l]; return; } int mid = (l + r) >> 1; build(x << 1,l,mid);build(x << 1 | 1,mid + 1,r); pushup(x); } inline void update(int x,int l,int r,int L,int R,LL v) { if(L <= l && r <= R) { tag[x] += v; seg[x] += (r - l + 1) * v; return; } pushdown(x,l,r); int mid = (l + r) >> 1; if(L <= mid)update(x << 1,l,mid,L,R,v); if(R > mid)update(x << 1 | 1,mid + 1,r,L,R,v); pushup(x); } inline LL query(int x,int l,int r,int L,int R) { if(L <= l && r <= R)return seg[x]; int mid = (l + r) >> 1;LL ans = 0; pushdown(x,l,r); if(L <= mid)ans += query(x << 1,l,mid,L,R); if(R > mid)ans += query(x << 1 | 1,mid + 1,r,L,R); return ans; } inline int isin(int x,int rt){return ind[rt] >= ind[x] && ind[rt] <= oud[x];} inline int getson(int x,int rt) { int t = dep[x] - dep[rt] - 1; for(int i=23;~i;i--)if(t & (1 << i))x = par[x][i]; return x; } inline void Modify(int x,LL v) { if(x == rt)update(1,1,n,1,n,v); else if(isin(x,rt)) { int vn = getson(rt,x); if(ind[vn] > 1)update(1,1,n,1,ind[vn] - 1,v); if(oud[vn] < n)update(1,1,n,oud[vn] + 1,n,v); } else update(1,1,n,ind[x],oud[x],v); } inline double Query(int x) { if(x == rt)return 1.0 * query(1,1,n,1,n) / n; else if(isin(x,rt)) { double ans = 0; int vn = getson(rt,x); if(ind[vn] > 1)ans += query(1,1,n,1,ind[vn] - 1); if(oud[vn] < n)ans += query(1,1,n,oud[vn] + 1,n); return ans / (n - size[vn]); } else return (1.0 * query(1,1,n,ind[x],oud[x]) / size[x]); } #define ls ch[x][0] #define rs ch[x][1] int fa[maxn],ch[maxn][2],rev[maxn],st[maxn],top; inline int isroot(int x){return ch[fa[x]][1] != x && ch[fa[x]][0] != x;} inline void pushdown(int x) { if(rev[x]) { swap(ls,rs); rev[ls] ^= 1; rev[rs] ^= 1; rev[x] ^= 1; } } inline void rotate(int x) { int y = fa[x],z = fa[y]; int l = (ch[y][1] == x),r = l ^ 1; if(!isroot(y))ch[z][ch[z][1] == y] = x; fa[x] = z;fa[ch[x][r]] = y;fa[y] = x; ch[y][l] = ch[x][r];ch[x][r] = y; } inline void splay(int x) { st[top = 1] = x; for(int i=x;!isroot(i);i=fa[i])st[++top] = fa[i]; for(int i=top;i;i--)pushdown(st[i]); while(!isroot(x)) { int y = fa[x],z = fa[y]; if(!isroot(y)) { if(ch[z][0] == y ^ ch[y][0] == x)rotate(x); else rotate(y); } rotate(x); } } inline int fd(int x) { pushdown(x); while(ls)x = ls,pushdown(x); return x; } inline void access(int x) { for(int y=0;x;rs=y,y=x,x=fa[x]) { splay(x); int tmp = fd(rs);if(rs) Modify(tmp,1); tmp = fd(y);if(y) Modify(tmp,-1); } } inline void makeroot(int x) { access(x); splay(x); rev[x] ^= 1; rt = x; } int main() { //freopen("20.in","r",stdin); //freopen("gen.out","w",stdout); n = read(),m = read();rt = 1; for(int i=2;i<=n;i++)ins(read(),read()); dep[1] = 1;dfs(1); build(1,1,n); for(int i=1;i<=n;i++)fa[i] = par[i][0]; char opt[50]; while(m--) { scanf("%s",opt + 1);int x = read(); if(opt[3] == 'L')access(x); else if(opt[3] == 'C')makeroot(x); else printf("%.10lf\n",Query(x)); } }
View Code