博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
[BZOJ2588]Count on a tree(LCA+主席树)
阅读量:5046 次
发布时间:2019-06-12

本文共 2531 字,大约阅读时间需要 8 分钟。

题面

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

分析

第i棵主席树存储dfs序(dfn)为i的节点到根的链上序列,按dfn更新

查询用dfn[x]+dfn[y]-dfn[lca(x,y)]-dfn[fa[lca(x,y)]](指对应的主席树相减 )

不能直接减去dfn[lca(x,y)]*2,因为lca那个点也算,要少减一次

代码

#include
#include
#include
#include
#include
#define maxn 100005#define maxlogn 20using namespace std;struct node{#ifdef DEBUG int l; int r;#endif int ls; int rs; int cnt;}tree[maxn*maxlogn];int root[maxn];int ptr;void push_up(int x){ tree[x].cnt=tree[tree[x].ls].cnt+tree[tree[x].rs].cnt;}void update(int &x,int last,int upos,int l,int r){ x=++ptr; tree[x]=tree[last];#ifdef DEBUG tree[x].l=l; tree[x].r=r; #endif if(l==r){ tree[x].cnt++; return; } int mid=(l+r)>>1; if(upos<=mid) update(tree[x].ls,tree[last].ls,upos,l,mid); else update(tree[x].rs,tree[last].rs,upos,mid+1,r); push_up(x);} int query(int x,int y,int lc,int lcfa,int k,int l,int r){ if(l==r) return l; int mid=(l+r)>>1; int lcnt=tree[tree[x].ls].cnt+tree[tree[y].ls].cnt-tree[tree[lc].ls].cnt-tree[tree[lcfa].ls].cnt; if(k<=lcnt) return query(tree[x].ls,tree[y].ls,tree[lc].ls,tree[lcfa].ls,k,l,mid); else return query(tree[x].rs,tree[y].rs,tree[lc].rs,tree[lcfa].rs,k-lcnt,mid+1,r); }struct edge{ int from; int to; int next;}E[maxn<<1];int sz=1;int head[maxn];void add_edge(int u,int v){ sz++; E[sz].from=u; E[sz].to=v; E[sz].next=head[u]; head[u]=sz;}int tim;int logn;int dfn[maxn];int hash_dfn[maxn];int anc[maxn][maxlogn];int deep[maxn];void dfs(int x,int fa){ dfn[x]=++tim; hash_dfn[tim]=x; anc[x][0]=fa; deep[x]=deep[fa]+1; for(int i=1;i<=logn;i++) anc[x][i]=anc[anc[x][i-1]][i-1]; for(int i=head[x];i;i=E[i].next){ int y=E[i].to; if(y!=fa){ dfs(y,x); } }}int lca(int x,int y){ if(deep[x]
=0;i--){ if(deep[anc[x][i]]>=deep[y]){ x=anc[x][i]; } } if(x==y) return x; for(int i=logn;i>=0;i--){ if(anc[x][i]!=anc[y][i]){ x=anc[x][i]; y=anc[y][i]; } } return anc[x][0];}int n,m;int dis_cnt;int val[maxn];int tmp[maxn];int main(){ int u,v,k; scanf("%d %d",&n,&m); int lastans=0; logn=log2(n); for(int i=1;i<=n;i++){ scanf("%d",&val[i]); tmp[i]=val[i]; } sort(tmp+1,tmp+1+n); dis_cnt=unique(tmp+1,tmp+1+n)-tmp-1; for(int i=1;i<=n;i++) val[i]=lower_bound(tmp+1,tmp+1+dis_cnt,val[i])-tmp; for(int i=1;i

转载于:https://www.cnblogs.com/birchtree/p/10851749.html

你可能感兴趣的文章
《梦断代码》读书笔记(三)
查看>>
FreeMarker解析json数据
查看>>
Java8 Lambda表达应用 -- 单线程游戏server+异步数据库操作
查看>>
次序+“选择不重复的记录”(3)——最大记录
查看>>
Codeforces 450 C. Jzzhu and Chocolate
查看>>
[Unity3D]Unity3D游戏开发MatchTarget的作用攀登效果实现
查看>>
ACdream 1115 Salmon And Cat (找规律&amp;&amp;打表)
查看>>
JSON、JSONP、Ajax的区别
查看>>
AngularJS学习篇(一)
查看>>
【转载】 IP实时传输协议RTP/RTCP详解
查看>>
关于Xshell无法连接centos6.4的问题
查看>>
Linux系统的数据写入机制--延迟写入
查看>>
css3动画——基本准则
查看>>
javaweb常识
查看>>
Java注解
查看>>
时间>金钱
查看>>
元数据元素
查看>>
Visual Studio Code 构建C/C++开发环境
查看>>
web自己主动保存表单
查看>>
一个小的日常实践——高速Fibonacci数算法
查看>>