1. 程式人生 > >BZOJ1103 [POI2007]大都市

BZOJ1103 [POI2007]大都市

可用 iss algo rip highlight swap -1 const ctype

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

5
1 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

2
1
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]大都市