1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
   | #include <iostream> #include <cmath> #include <queue>  #define maxn 100010 #define sqrtn 400 using namespace std;
  int n, m;
  struct Edge {     int to, next; } e[maxn * 2]; int c1, head[maxn]; inline void add_edge(int u, int v) {     e[c1].to = v; e[c1].next = head[u]; head[u] = c1++; }
  int id[maxn * 2], in[maxn], dep[maxn]; void dfs(int u, int fa) {     static int cnt = 0;      id[++cnt] = u; in[u] = cnt;     for (int i = head[u]; ~i; i = e[i].next) {         int v = e[i].to; if (v == fa) continue;         dep[v] = dep[u] + 1; dfs(v, u); id[++cnt] = u;     }  }
  inline int st_min(int l, int r) { return in[l] < in[r] ? l : r; }
  int st[maxn * 2][21], Log[maxn * 2]; void init_st(int n) { Log[0] = -1;     for (int i = 1; i <= n; ++i) Log[i] = Log[i >> 1] + 1;     for (int i = 1; i <= n; ++i) st[i][0] = id[i];     for (int j = 1; j <= 20; ++j)         for (int i = 1; i + (1 << j) - 1 <= n; ++i)             st[i][j] = st_min(st[i][j - 1], st[i + (1 << j - 1)][j - 1]);  }
  inline int get_lca(int l, int r) {     l = in[l]; r = in[r]; if (l > r) swap(l, r);     int k = Log[r - l + 1];     return st_min(st[l][k], st[r - (1 << k) + 1][k]); }
  inline int D(int x, int y) { return dep[x] + dep[y] - 2 * dep[get_lca(x, y)]; }
  int blo, num, l[sqrtn], r[sqrtn], bl[maxn]; void init_blo(int n) {     blo = sqrt(n); num = (n + blo - 1) / blo;      for (int i = 1; i <= n; ++i) bl[i] = (i - 1) / blo + 1;     for (int i = 1; i <= num; ++i) {         l[i] = (i - 1) * blo + 1;         r[i] = i * blo;     } r[num] = n;  }
  struct Query {     int opt, x; } Q[maxn];
  int ans[maxn];
  int dis[maxn]; bool vis[maxn]; void bfs(int k) {     fill(dis, dis + maxn, 0);      fill(vis, vis + maxn, 0);      queue<int> Q;     for (int i = l[k]; i <= r[k]; ++i) {         int opt = ::Q[i].opt, x = ::Q[i].x;         if (opt == 1) {             dis[x] = 0; vis[x] = 1;             Q.push(x);         }     }     while (!Q.empty()) {         int u = Q.front(); Q.pop();         if (dis[u] < ans[u]) ans[u] = dis[u];          for (int i = head[u]; ~i; i = e[i].next) {             int v = e[i].to; if (vis[v]) continue;             vis[v] = 1; dis[v] = dis[u] + 1; Q.push(v);         }     } } 
  int main() { fill(head, head + maxn, -1);      ios::sync_with_stdio(false);     cin.tie(nullptr); cout.tie(nullptr);
      cin >> n >> m;      for (int i = 1; i < n; ++i) {         int x, y; cin >> x >> y;         add_edge(x, y); add_edge(y, x);     } dep[1] = 1; dfs(1, 0); init_st(2 * n - 1); init_blo(m); bl[0] = 1;     for (int i = 1; i <= n; ++i) ans[i] = D(i, 1);      for (int i = 1; i <= m; ++i) {         int opt, x; cin >> opt >> x; Q[i] = { opt, x };         if (bl[i] != bl[i - 1]) bfs(bl[i - 1]);         if (opt == 2) {             int ans = ::ans[x];             for (int j = l[bl[i]]; j < i; ++j)                 if (Q[j].opt == 1) ans = min(ans, D(x, Q[j].x));             cout << ans << "\n";          }     }      return 0; }  
 
   |