BZOJ1103 [POI2007]大都市
Description
在經濟全球化浪潮的影響下,習慣於漫步在清晨的鄉間小路的郵遞員Blue Mary也開始騎著摩托車傳遞郵件了。
不過,她經常回憶起以前在鄉間漫步的情景。昔日,鄉下有依次編號為1..n的n個小村莊,某些村莊之間有一些雙
向的土路。從每個村莊都恰好有一條路徑到達村莊1(即比特堡)。並且,對於每個村莊,它到比特堡的路徑恰好
只經過編號比它的編號小的村莊。另外,對於所有道路而言,它們都不在除村莊以外的其他地點相遇。在這個未開
化的地方,從來沒有過高架橋和地下鐵道。隨著時間的推移,越來越多的土路被改造成了公路。至今,Blue Mary
還清晰地記得最後一條土路被改造為公路的情景。現在,這裏已經沒有土路了——所有的路都成為了公路,而昔日
的村莊已經變成了一個大都市。 Blue Mary想起了在改造期間她送信的經歷。她從比特堡出發,需要去某個村莊,
並且在兩次送信經歷的間隔期間,有某些土路被改造成了公路.現在Blue Mary需要你的幫助:計算出每次送信她需
要走過的土路數目。(對於公路,她可以騎摩托車;而對於土路,她就只好推車了。)
Input
第一行是一個數n(1 < = n < = 2 50000).以下n-1行,每行兩個整數a,b(1 < = a以下一行包含一個整數m
(1 < = m < = 2 50000),表示Blue Mary曾經在改造期間送過m次信。以下n+m-1行,每行有兩種格式的若幹信息
,表示按時間先後發生過的n+m-1次事件:若這行為 A a b(a若這行為 W a, 則表示Blue Mary曾經從比特堡送信到
村莊a。
Output
有m行,每行包含一個整數,表示對應的某次送信時經過的土路數目。
Sample Input
51 2
1 3
1 4
4 5
4
W 5
A 1 4
W 5
A 4 5
W 5
W 2
A 1 2
A 1 3
Sample Output
21
0
1
題解
可以發現最初的每個節點的答案即為它的深度,每次將節點k和他的父親之間的邊改建為公路時其子樹中所有答案減一。
那麽可以求出dfs序,子樹答案減一即為區間減,詢問即為單點查詢,可用樹狀數組維護。
代碼中由於每個節點的父親編號比其自身編號小,逆序枚舉節點,將其siz+1,就求出了其大小,此時其父節點的siz就是它的兄弟中排在他前面的所有子樹的siz之和,也就是其在dfs序中比他的父親靠後多少。再將其大小加到其父節點的siz裏,最後順序枚舉一遍即可。
附代碼:
#include <algorithm> #include <cctype> #include <cstdio> const int N = 250050; inline int readInt() { int ans = 0; char c; do c = getchar(); while (!isdigit(c)); do { ans = ans * 10 + c - ‘0‘; c = getchar(); } while (isdigit(c)); return ans; } inline char readChar() { char c; do c = getchar(); while (isspace(c)); return c; } int left[N], siz[N], fa[N]; int A[N], n; inline void add(int i, int x) { while (i <= n) { A[i] += x; i += i & -i; } } inline int query(int i) { int ans = 0; while (i) { ans += A[i]; i -= i & -i; } return ans; } int main() { n = readInt(); for (int i = 1; i < n; ++i) { int a = readInt(), b = readInt(); if (a > b) std::swap(a, b); fa[b] = a; } fa[1] = 0; for (int i = n; i; --i) { left[i] = siz[fa[i]] + 1; siz[fa[i]] += ++siz[i]; } for (int i = 1; i <= n; ++i) left[i] += left[fa[i]]; for (int i = 2; i <= n; ++i) { add(left[i], 1); add(left[i] + siz[i], -1); } int m = readInt() + n - 1; while (m--) { if (readChar() == ‘W‘) printf("%d\n", query(left[readInt()])); else { int a = std::max(readInt(), readInt()); add(left[a], -1); add(left[a] + siz[a], 1); } } return 0; }
BZOJ1103 [POI2007]大都市