题意:

给\(n(1 \leq n \leq 10^5)\)一棵树,每个点有个权值。
还有\(m(1 \leq m \leq 10^5)\)个询问:
\(u \, v \, k\),查询路径\(u \to v\)上权值第\(k\)小的权值。

分析:

和普通的区间一样,对于树维护到根节点的路径信息,父亲节点代表的树就是它的前一棵树。
查询的时候还要求出\(LCA(u,v)\),路径上的点数就是:

  • \(u\)到根的个数+\(v\)到根的个数-\(lca\)到根的个数\(\times 2\)

如果\(lca\)的权值也在统计范围内,统计个数再加\(1\)。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;const int maxn = 100000 + 10;struct Edge
{int v, nxt;Edge() {}Edge(int v, int nxt): v(v), nxt(nxt) {}
};int head[maxn], ecnt;
Edge edges[maxn * 2];void AddEdge(int u, int v) {edges[ecnt] = Edge(v, head[u]);head[u] = ecnt++;
}int fa[maxn], L[maxn];void dfs(int u) {for(int i = head[u]; ~i; i = edges[i].nxt) {int v = edges[i].v;if(v == fa[u]) continue;fa[v] = u;L[v] = L[u] + 1;dfs(v);}
}int n, m;int anc[maxn][20];void preprocess() {memset(anc, 0, sizeof(anc));for(int i = 1; i <= n; i++) anc[i][0] = fa[i];for(int j = 1; (1 << j) < n; j++)for(int i = 1; i <= n; i++) if(anc[i][j-1])anc[i][j] = anc[anc[i][j-1]][j-1];
}int LCA(int u, int v) {if(L[u] < L[v]) swap(u, v);int log;for(log = 0; (1 << log) < L[u]; log++);for(int i = log; i >= 0; i--)if(L[u] - (1<<i) >= L[v]) u = anc[u][i];if(u == v) return u;for(int i = log; i >= 0; i--)if(anc[u][i] && anc[u][i] != anc[v][i])u = anc[u][i], v = anc[v][i];return fa[u];
}int a[maxn], b[maxn], tot;struct Node
{int lch, rch, sum;
};int sz;
Node T[maxn << 5];
int root[maxn];int newnode() {sz++;T[sz].lch = T[sz].rch = T[sz].sum = 0;return sz;
}int update(int pre, int L, int R, int p) {int rt = newnode();T[rt].lch = T[pre].lch;T[rt].rch = T[pre].rch;T[rt].sum = T[pre].sum + 1;if(L == R) return rt;int M = (L + R) / 2;if(p <= M) T[rt].lch = update(T[pre].lch, L, M, p);else T[rt].rch = update(T[pre].rch, M+1, R, p);return rt;
}void dfs2(int u) {root[u] = update(root[fa[u]], 1, tot, a[u]);for(int i = head[u]; ~i; i = edges[i].nxt) {int v = edges[i].v;if(v == fa[u]) continue;dfs2(v);}
}int tmp;int query(int u, int v, int l, int L, int R, int k) {if(L == R) return L;int M = (L + R) / 2;int sum = T[T[u].lch].sum + T[T[v].lch].sum - T[T[l].lch].sum * 2;if(L <= tmp && tmp <= M) sum++;if(sum >= k) return query(T[u].lch, T[v].lch, T[l].lch, L, M, k);else return query(T[u].rch, T[v].rch, T[l].rch, M+1, R, k - sum);
}int main()
{scanf("%d%d", &n, &m);for(int i = 1; i <= n; i++) {scanf("%d", a + i);b[i] = a[i];}ecnt = 0;memset(head, -1, sizeof(head));for(int i = 1; i < n; i++) {int u, v; scanf("%d%d", &u, &v);AddEdge(u, v); AddEdge(v, u);}dfs(1);preprocess();sort(b + 1, b + 1 + n);tot = unique(b + 1, b + 1 + n) - b - 1;for(int i = 1; i <= n; i++)a[i] = lower_bound(b + 1, b + 1 + tot, a[i]) - b;dfs2(1);while(m--) {int u, v, k; scanf("%d%d%d", &u, &v, &k);int lca = LCA(u, v);tmp = a[lca];int ans = query(root[u], root[v], root[lca], 1, tot, k);printf("%d\n", b[ans]);}return 0;
}

转载于:https://www.cnblogs.com/AOQNRMGYXLMV/p/5285890.html

SPOJ COT Count on a tree 主席树相关推荐

  1. SPOJ 10628 Count on a tree 主席树 附数据生成器

    很奇怪的题啊,感觉思路和别人一样,但是我得用快速IO才能AC--不然就T 没用快速output,只用了快速input 而且居然限制代码长度...代码要短于6000B,我改了好久啊 题目大意:给一棵树, ...

  2. bzoj2588: Spoj 10628. Count on a tree 主席树

    在每一个点的父亲做主席树,每次访问时用两个点的和减去其LCA和LCA父亲的和即可. #include<bits/stdc++.h> using namespace std; inline ...

  3. SPOJ - COT Count on a tree [LCA+主席树]【数据结构】

    题目链接:http://www.spoj.com/problems/COT/en/ -------------------------------------- COT - Count on a tr ...

  4. SPOJ - COT Count on a tree(LCA+主席树+离散化)

    题目链接:点击查看 题目大意:给出一棵树,每个点都有一个权值,现在给出m个询问,每次询问的格式是u,v,k,要求输出u-v这条路径上第k大的数 题目分析:一看到第k大的数就会想到主席树,既然是在树上的 ...

  5. SPOJ COT Count on a tree(主席树+倍增lca)

    思路:这个题其实就是树上的第k小,主席树的本质还是类似于前缀和一样的结构,所以是完全相同的,所以我们在树上也可以用同样的方法,我们对于每一个节点进行建树,然后和普通的树上相同,ab之间的距离是等于 r ...

  6. BZOJ 2588: Spoj 10628. Count on a tree 树上跑主席树

    2588: Spoj 10628. Count on a tree Time Limit: 1 Sec Memory Limit: 256 MB 题目连接 http://www.lydsy.com/J ...

  7. BZOJ 2588: Spoj 10628. Count on a tree( LCA + 主席树 )

    Orz..跑得还挺快的#10 自从会树链剖分后LCA就没写过倍增了... 这道题用可持久化线段树..点x的线段树表示ROOT到x的这条路径上的权值线段树 ----------------------- ...

  8. bzoj 2588 Spoj 10628. Count on a tree (可持久化线段树)

    Spoj 10628. Count on a tree Time Limit: 12 Sec  Memory Limit: 128 MB Submit: 7669  Solved: 1894 [Sub ...

  9. 【SPOJ】Count On A Tree II(树上莫队)

    [SPOJ]Count On A Tree II(树上莫队) 题面 洛谷 Vjudge 洛谷上有翻译啦 题解 如果不在树上就是一个很裸很裸的莫队 现在在树上,就是一个很裸很裸的树上莫队啦. #incl ...

最新文章

  1. 了解下RDF 容器元素
  2. Android 线程池概念及使用
  3. Basic操作系统概念梳理
  4. java list详解_Java 中 list 用法案例详解
  5. [Windows Phone 7] Storage
  6. HttpClient 使用证书访问https站点
  7. 如何选择jquery版本
  8. 【语音识别】日语语音识别系统Julius(v4.4)的基于DNN的识别(5月8号:识别结果更新)
  9. 通信协议(一)——UART协议
  10. php addslashes 防注入,PHP基础-addslashes防sql注入-北漂者
  11. 谷歌账号--手机号无法验证
  12. 取模(余)%运算详解
  13. 详解项目管理中任务、成本、产品三者的关系
  14. 未能加载文件或程序集 System Drawing Version 4 0 0 0 Culture neutral
  15. 程序员之路:Linux目录处理命令_彭世瑜_新浪博客
  16. LVGL8.2移植学习
  17. 被“短信嗅探”盯上 一觉醒来一无所有?
  18. 数据模型:概念数据模型,逻辑数据模型,物理数据模型
  19. PMOS管和NMOS管
  20. 《抓住听众心理——演讲者要知道的100件事》一第 1 章 人们是怎样思考和学习的...

热门文章

  1. resource fork, Finder information, or similar detr
  2. Oracle中row_number()、rank()、dense_rank() 的区别
  3. docker-ce安装
  4. 使用Server 2008新GPO做驱动器映射
  5. mysql 表锁-解锁
  6. 计算机学科技术前沿:互联网上信息可信性的现状
  7. POJ 2976 Dropping Tests
  8. 继续泼冷水 Ubuntu决定放弃支持安腾
  9. Android Studio出现Failed to open zip file. Gradle's dependency cache may be corrupt问题的解决
  10. java 深入了解DTO及如何使用DTO