BZOJ 2588: Spoj 10628. Count on a tree

2588: Spoj 10628. Count on a tree

Time Limit: 12 Sec  Memory Limit: 128 MB
Submit: 5394  Solved: 1272
[Submit][Status][Discuss]

Description

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v,k)，你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案，初始为0，即第一个询问的u是明文。

 

 

Input

第一行两个整数N,M。

第二行有N个整数，其中第i个整数表示点i的权值。

后面N-1行每行两个整数(x,y)，表示点x到点y有一条边。

最后M行每行两个整数(u,v,k)，表示一组询问。

 

Output

 

M行，表示每个询问的答案。

Sample Input

8 5
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5 1
0 5 2
10 5 3
11 5 4
110 8 2

Sample Output

2
8
9
105
7

HINT

 

HINT：

N,M<=100000

暴力自重。。。

 

Source

鸣谢seter

[Submit][Status][Discuss]

 

树上主席树

 

#include <bits/stdc++.h>
const int siz = 200005, tre = 7000005;
int n, m, q, val[siz], map[siz], tot, hd[siz], to[siz], nt[siz], edge, root[siz], ls[tre], rs[tre], sm[tre], cnt, vis[siz], fa[siz][21], dep[siz], qa[siz], qb[siz], tail;
void insert(int &t, int f, int l, int r, int p) {
    t = ++cnt; ls[t] = ls[f], rs[t] = rs[f], sm[t] = sm[f] + 1;
    if (l != r) { int mid = (l + r) >> 1; (p <= mid ? insert(ls[t], ls[f], l, mid, p) : insert(rs[t], rs[f], mid + 1, r, p)); }
}
void dfs(int u, int f) { vis[u] = 1; 
    dep[u] = dep[f] + 1, fa[u][0] = f;
    insert(root[u], root[f], 1, tot, val[u]);
    for (int i = 1; i <= 20; ++i)fa[u][i] = fa[fa[u][i - 1]][i - 1];
    for (int i = hd[u]; i; i = nt[i])if (to[i] != f)dfs(to[i], u);
}
inline int lca(int a, int b) {
    if (dep[a] < dep[b])a ^= b ^= a ^= b;
    for (int i = 20; ~i; --i)if (dep[fa[a][i]] >= dep[b])a = fa[a][i];
    for (int i = 20; ~i; --i)if (fa[a][i] ^ fa[b][i])a = fa[a][i], b = fa[b][i];
    return a == b ? a : fa[a][0];
}
inline void add(int a, int b) { qa[tail] = root[a], qb[tail++] = b; }
int qry(int l, int r, int k) { if (l == r)return l; int s = 0, mid = (l + r) >> 1; 
    for (int i = 0; i < tail; ++i)s += qb[i] * sm[ls[qa[i]]];
    for (int i = 0; i < tail; ++i)qa[i] = s < k ? rs[qa[i]] : ls[qa[i]];
    return s < k ? qry(mid + 1, r, k - s) : qry(l, mid, k);
}
inline void adde(int x, int y) {
    nt[++edge] = hd[x], to[edge] = y, hd[x] = edge,
    nt[++edge] = hd[y], to[edge] = x, hd[y] = edge;
}
signed main(void) {
    scanf("%d%d", &n, &q); m = n - 1;
    for (int i = 1; i <= n; ++i)scanf("%d", val + i), map[++tot] = val[i];
    std::sort(map + 1, map + tot + 1); tot = std::unique(map + 1, map + tot + 1) - map;
    for (int i = 1; i <= n; ++i)val[i] = std::lower_bound(map + 1, map + tot, val[i]) - map;
    for (int i = 1, x, y; i <= m; ++i)scanf("%d%d", &x, &y), adde(x, y);
    for (int i = 1; i <= n; ++i)if (!vis[i])dfs(i, 0);
    for (int i = 1, ans = 0, x, y, k, t; i <= q; ++i) { 
        scanf("%d%d%d", &x, &y, &k), x ^= ans, tail = 0, 
        add(x, 1), add(y, 1), add(t = lca(x, y), -1), add(fa[t][0], -1), printf("%d", ans = map[qry(1, tot, k)]); if (i != q)puts("");
    }
}

 

@Author: YouSiki