1. 程式人生 > >[BZOJ 2588] Count on a tree

[BZOJ 2588] Count on a tree

[題目連結]

           https://www.lydsy.com/JudgeOnline/problem.php?id=2588

[演算法]

         如果我們能知道“u到v這條路徑上權值<= k的數的個數” ,  那麼就可以通過二分的方式求出答案

         進一步地 , u到v路徑上權值<= k的數的個數 = u到根節點路徑上權值<= k的數的個數 + v到根節點路徑上權值<= k的數的個數 - u和v的最近公共祖先到根節點路徑上權值<= k的數的個數 - u和v的最近公共祖先的父節點到根節點路徑上權值<= k的數的個數 

         建立可持久化線段樹 , 查詢時線上段樹上二分即可

         時間複雜度 : O(NlogN)

[程式碼]

        

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010
#define MAXLOG 20
typedef long long ll;
typedef long double ld;

struct edge
{
        
int to , nxt; } e[MAXN << 1]; int n , q , len , idx , tot; int lson[MAXN * 20] , rson[MAXN * 20] , sum[MAXN * 20] , depth[MAXN] , head[MAXN] , root[MAXN] , son[MAXN] , size[MAXN] , fa[MAXN] , top[MAXN]; int a[MAXN] , tmp[MAXN]; int up[MAXN][MAXLOG]; template <typename T> inline void chkmax(T &x,T y) { x = max(x,y); } template
<typename T> inline void chkmin(T &x,T y) { x = min(x,y); } template <typename T> inline void read(T &x) { T f = 1; x = 0; char c = getchar(); for (; !isdigit(c); c = getchar()) if (c == '-') f = -f; for (; isdigit(c); c = getchar()) x = (x << 3) + (x << 1) + c - '0'; x *= f; } inline void addedge(int u , int v) { ++tot; e[tot] = (edge){v , head[u]}; head[u] = tot; } inline void build(int &k , int l , int r) { k = ++idx; if (l == r) return; int mid = (l + r) >> 1; build(lson[k] , l , mid); build(rson[k] , mid + 1 , r); } inline void modify(int &k , int old , int l , int r , int pos , int value) { k = ++idx; lson[k] = lson[old] , rson[k] = rson[old]; sum[k] = sum[old] + value; if (l == r) return; int mid = (l + r) >> 1; if (mid >= pos) modify(lson[k] , lson[k] , l , mid , pos , value); else modify(rson[k] , rson[k] , mid + 1 , r , pos , value); } inline int query(int rt1 , int rt2 , int rt3 , int rt4 , int l , int r , int k) { if (l == r) return l; int value = sum[lson[rt1]] + sum[lson[rt2]] - sum[lson[rt3]] - sum[lson[rt4]]; int mid = (l + r) >> 1; if (value >= k) return query(lson[rt1] , lson[rt2] , lson[rt3] , lson[rt4] , l , mid , k); else return query(rson[rt1] , rson[rt2] , rson[rt3] , rson[rt4] , mid + 1 , r , k - value); } inline void dfs1(int u , int father) { size[u] = 1; depth[u] = depth[father] + 1; modify(root[u] , root[father] , 1 , len , a[u] , 1); for (int i = head[u]; i; i = e[i].nxt) { int v = e[i].to; if (v == father) continue; dfs1(v , u); fa[v] = u; size[u] += size[v]; if (son[u] == 0 || size[v] > size[son[u]]) son[u] = v; } } inline void dfs2(int u , int tp) { top[u] = tp; if (son[u]) dfs2(son[u] , tp); for (int i = head[u]; i; i = e[i].nxt) { int v = e[i].to; if (v != son[u] && v != fa[u]) dfs2(v , v); } } inline int lca(int x , int y) { while (top[x] != top[y]) { if (depth[top[x]] > depth[top[y]]) swap(x , y); y = fa[top[y]]; } if (depth[x] < depth[y]) return x; else return y; } inline int getans(int u , int v , int k) { int t1 = lca(u , v) , t2 = fa[t1]; return query(root[u] , root[v] , root[t1] , root[t2] , 1 , len , k); } int main() { read(n); read(q); for (int i = 1; i <= n; i++) { read(a[i]); tmp[i] = a[i]; } for (int i = 1; i < n; i++) { int x , y; read(x); read(y); addedge(x , y); addedge(y , x); } sort(tmp + 1 , tmp + n + 1); len = unique(tmp + 1 , tmp + n + 1) - tmp - 1; for (int i = 1; i <= n; i++) a[i] = lower_bound(tmp + 1 , tmp + len + 1 , a[i]) - tmp; build(root[0] , 1 , len); dfs1(1 , 0); dfs2(1 , 1); int lastans = 0; while (q--) { int u , v , k; read(u); read(v); read(k); u ^= lastans; printf("%d\n" , lastans = tmp[getans(u , v , k)]); } return 0; }