【BZOJ2588】Count on a tree，主席树维护链+ST表求LCA

传送门
写在前面：一天下来就写了两道主席树的题……（codevs上的一道智障天梯不算）
思路：
才知道原来主席树不仅可以通过dfs序维护子树区间，还可以直接维护一条到根的链……
我们建好主席树后，每次查询u->v路径上的第k大，无非有两种情况
1.u,v在同一条链上
2.u,v不在同一条链上
其实这两种情况处理起来是一样的，我们利用主席树中的前缀和思路，root[u]并上root[v]再减去root[lca(u,v)]再减去root[fa[lca(u,v)]]就是u->v路径上的点了（可以自己画图感受一下）
所以每次查询时要算四棵树之间的siz关系
注意：
1.求lca我用的是ST表，当然倍增也是可以的
2.最后一个答案不要换行！
代码：

#include<bits/stdc++.h>
#define M 100004
#define ls(x) a[x].ch[0]
#define rs(x) a[x].ch[1]
using namespace std;
int n,m,lastans,tot,maxn,cnt;
int fa[M],w[M],ID[M],first[M],dep[M],ST[M<<1][20],pos[M],num[M<<1];
struct disc
{int data,id;bool operator <(const disc other)const{return data<other.data;}
}d[M];
struct edge
{int u,v,next;
}e[M<<1];
struct Chairman_tree
{int siz,ch[2];
}a[M*20];
int in()
{char ch=getchar();int t=0;while (!isdigit(ch))ch=getchar();while (isdigit(ch)) t=(t<<3)+(t<<1)+ch-48,ch=getchar();return t;
}
void add(int x,int y)
{e[++tot]=(edge){x,y,first[x]};first[x]=tot;e[++tot]=(edge){y,x,first[y]};first[y]=tot;
}
void build(int now,int L,int R,int rt,int val)
{a[rt].siz=a[now].siz+1;if (L==R) return;int mid=(L+R)>>1;if (mid>=val)rs(rt)=rs(now),ls(rt)=++cnt,build(ls(now),L,mid,ls(rt),val);elsea[rt].ch[0]=a[now].ch[0],a[rt].ch[1]=++cnt,build(rs(now),mid+1,R,rs(rt),val);
}
void dfs(int x)
{num[++num[0]]=x;pos[x]=num[0];ST[num[0]][0]=x;if (x==1) build(1,1,maxn,2,w[x]);for (int i=first[x];i;i=e[i].next)if (!pos[e[i].v]){dep[e[i].v]=dep[x]+1;fa[e[i].v]=x;build(x+1,1,maxn,e[i].v+1,w[e[i].v]);dfs(e[i].v);num[++num[0]]=x;ST[num[0]][0]=x;}
}
void init()
{int x,y;for (int i=1;i<=log2(num[0]);i++)for (int j=1;j<=num[0]-(1<<i)+1;j++)x=ST[j][i-1],y=ST[j+(1<<i-1)][i-1],ST[j][i]=(dep[x]>dep[y]?y:x);
}
int LCA(int x,int y)
{if (pos[x]>pos[y]) swap(x,y);int p=pos[x],q=pos[y],t=log2(q-p+1);if (dep[ST[p][t]]>dep[ST[q-(1<<t)+1][t]]) return ST[q-(1<<t)+1][t];else return ST[p][t];
}
int get(int begin,int end,int u,int v,int lca,int fa,int k)
{if (begin==end) return end;int mid=(begin+end)>>1;int t=a[ls(u)].siz+a[ls(v)].siz-a[ls(lca)].siz-a[ls(fa)].siz;if (k>t)return get(mid+1,end,rs(u),rs(v),rs(lca),rs(fa),k-t);elsereturn get(begin,mid,ls(u),ls(v),ls(lca),ls(fa),k);
}
main()
{n=in();m=in();for (int i=1;i<=n;i++)w[i]=in(),d[i].id=i,d[i].data=w[i];sort(d+1,d+n+1);maxn=1;ID[1]=d[1].data;w[d[1].id]=1;for (int i=2;i<=n;i++){if (d[i].data!=d[i-1].data) maxn++,ID[maxn]=d[i].data;w[d[i].id]=maxn;}for (int i=1;i<n;i++)add(in(),in());cnt=n+1;dfs(1);init();int x,y,z,k;while (m--){x=in()^lastans;y=in();k=in();z=LCA(x,y);lastans=ID[get(1,maxn,x+1,y+1,z+1,fa[z]+1,k)];printf("%d",lastans);if (m) putchar('\n');}
}

【BZOJ2588】Count on a tree，主席树维护链+ST表求LCA相关推荐

bzoj2588: Spoj 10628. Count on a tree 主席树
在每一个点的父亲做主席树,每次访问时用两个点的和减去其LCA和LCA父亲的和即可. #include<bits/stdc++.h> using namespace std; inline ...
SPOJ 10628 Count on a tree 主席树附数据生成器
很奇怪的题啊,感觉思路和别人一样,但是我得用快速IO才能AC--不然就T 没用快速output,只用了快速input 而且居然限制代码长度...代码要短于6000B,我改了好久啊题目大意:给一棵树, ...
SPOJ COT Count on a tree 主席树
题意: 给\(n(1 \leq n \leq 10^5)\)一棵树,每个点有个权值. 还有\(m(1 \leq m \leq 10^5)\)个询问: \(u \, v \, k\),查询路径\(u \ ...
BZOJ2588 Count on a tree DFS序+LCA+值域主席树
Count on a tree 给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权.其中lastans是上一个询问的答 ...
BZOJ2588 Count on a tree 【树上主席树】
2588: Spoj 10628. Count on a tree Time Limit: 12 Sec Memory Limit: 128 MB Submit: 7577 Solved: 185 ...
HDU - 3804 Query on a tree(主席树维护最大值+离散化)
题目链接:点击查看题目大意:给出一棵树,每条边上都有一个权值,给出m个查询:a,b:问从点1到点a的唯一路径上,在边权小于等于b的边中选出边权最大的值输出,若没有符合条件的边则输出-1: 题目分析: ...
[bzoj2588] Count on a tree
洛谷 P2633 传送门 bzoj 2588 传送门一道树上差分的题. 每个点都建一个权值线段树,维护的是从这个点到根的链的信息. 这样就可以用主席树了,每个点的版本由其父节点的版本加上该点的权值得 ...
ACM模板——线段树树状数组ST表
int bit[maxn],n; void init() {n = maxn;memset(bit,0,sizeof(bit)); } int sum(int i) {int s = 0;while( ...
POJ2823 Sliding Window【单调队列】【线段树】【ST表】
Sliding Window POJ - 2823 题意: 给出一个长度为N的序列,通过一个窗口,可以看到序列中连续的K个元素,窗口从最左边出发,每次移动一个单位,对于每次移动,输出当前窗口中的最大值 ...

【BZOJ2588】Count on a tree，主席树维护链+ST表求LCA

【BZOJ2588】Count on a tree，主席树维护链+ST表求LCA相关推荐

最新文章

热门文章