[BZOJ2588][Spoj10628]Count on a tree(树上主席树)
2017-12-17 16:17
555 查看
题目:
我是超链接题解:
给每个点建一棵主席树,维护到根节点的所有点,这样答案就是sum(a)+sum(b)−sum(lca(a,b))−sum(fa(lca(a,b)))代码:
#include <cstdio> #include <iostream> #include <algorithm> using namespace std; const int mc=18; const int N=100005; int tot,nxt[N*2],point ,v[N*2],h ,f [mc],dfn ,num,mi[mc]; int b ,root ,sz,a ,s,pos ; struct hh{int l,r,w;}t[N*mc*5]; void addline(int x,int y) { ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; } void dfs(int x,int fa) { h[x]=h[fa]+1;dfn[++num]=x; for (int i=1;i<mc;i++) if (mi[i]>h[x]) break; else f[x][i]=f[f[x][i-1]][i-1]; for (int i=point[x];i;i=nxt[i]) if (v[i]!=fa) { f[v[i]][0]=x; dfs(v[i],x); } } int lca(int x,int y) { if (h[x]<h[y]) swap(x,y); int k=h[x]-h[y]; for (int i=0;i<mc;i++) if (k&(1<<i)) x=f[x][i]; if (x==y) return x; for (int i=mc-1;i>=0;i--) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; return f[x][0]; } void insert(int &now,int l,int r,int x) { t[++sz]=t[now]; now=sz; t[now].w++; if (l==r) return; int mid=(l+r)>>1; if (x<=mid) insert(t[now].l,l,mid,x); else insert(t[now].r,mid+1,r,x); } int qurry(int l,int r,int x,int y,int fa,int of,int k) { if (l==r) return b[l]; int mid=(l+r)>>1; if (t[t[x].l].w+t[t[y].l].w-t[t[fa].l].w-t[t[of].l].w>=k) return qurry(l,mid,t[x].l,t[y].l,t[fa].l,t[of].l,k); else return qurry(mid+1,r,t[x].r,t[y].r,t[fa].r,t[of].r,k-(t[t[x].l].w+t[t[y].l].w-t[t[fa].l].w-t[t[of].l].w)); } int main() { int m,x,y,k,n,ans=0;mi[0]=1; for (int i=0;i<mc;i++) mi[i]=mi[i-1]*2; scanf("%d%d",&n,&m); for (int i=1;i<=n;i++) scanf("%d",&a[i]),b[++s]=a[i]; sort(b+1,b+s+1); s=unique(b+1,b+s+1)-b-1; for (int i=1;i<=n;i++) pos[i]=lower_bound(b+1,b+s+1,a[i])-b; for (int i=1;i<n;i++) { scanf("%d%d",&x,&y); addline(x,y); } dfs(1,0); for (int i=1;i<=num;i++) { root[dfn[i]]=root[f[dfn[i]][0]]; insert(root[dfn[i]],1,s,pos[dfn[i]]); } for (int i=1;i<=m;i++) { scanf("%d%d%d",&x,&y,&k); x^=ans; ans=qurry(1,s,root[x],root[y],root[lca(x,y)],root[f[lca(x,y)][0]],k); printf("%d",ans); if (i!=m) printf("\n"); } }
相关文章推荐
- BZOJ 2588: Spoj 10628. Count on a tree | 树上主席树
- BZOJ 2588: Spoj 10628. Count on a tree 树上跑主席树
- 【BZOJ 2588】 Spoj 10628. Count on a tree|树上K大|树链剖分|主席树
- BZOJ BZOJ 2588: Spoj 10628. Count on a tree 树上主席树
- bzoj2588: Spoj 10628. Count on a tree(树上第k大)(主席树)
- bzoj 2588: Spoj 10628. Count on a tree(树上主席树)
- BZOJ 2588 Spoj 10628 Count on a tree | 树上主席树
- BZOJ 2588: Spoj 10628. Count on a tree [树上主席树]
- bzoj2588 Spoj 10628. Count on a tree【主席树+lca】
- bzoj2588: Spoj 10628. Count on a tree 主席树+dfs序
- 【BZOJ2588】Spoj 10628. Count on a tree 主席树+LCA
- BZOJ 2588: Spoj 10628. Count on a tree( LCA + 主席树 )
- bzoj 2588: Spoj 10628. Count on a tree【主席树+倍增】
- bzoj 2588: Spoj 10628. Count on a tree LCA+主席树
- [bzoj2588][Spoj10628]Count on a tree_主席树
- 【BZOJ2588】Spoj 10628. Count on a tree【主席树】【LCA】
- BZOJ_2588_Spoj 10628. Count on a tree_树剖+主席树
- BZOJ 2588: Spoj 10628. Count on a tree 主席树+lca
- [主席树] BZOJ 2588 Spoj 10628. Count on a tree
- bzoj 2588 Spoj 10628. Count on a tree(主席树)