1. 程式人生 > >2018.10.06 spoj QTREE3(樹鏈剖分)

2018.10.06 spoj QTREE3(樹鏈剖分)

傳送門 發現把樹投射到序列上。 發現每個黑點都只會對其子樹產生影響。 再繼續想想好像在投射到序列上之後好像就是查詢從點uu到根路徑上深度最淺的黑點。 也就是loglog段線段樹上最靠左的黑點。 這個直接線上段樹上二分到葉子結點返回下標就行了。 程式碼:

#include<bits/stdc++.h>
#define N 100005
#define lc (p<<1)
#define rc (p<<1|1)
#define mid (T[p].l+T[p].r>>1) 
using namespace std;
inline int read(){
int ans=0; char ch=getchar(); while(!isdigit(ch))ch=getchar(); while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar(); return ans; } int n,m,cnt,tot,first[N],siz[N],hson[N],fa[N],pred[N],num[N],top[N],dep[N]; struct edge{int v,next;}e[N<<1]; struct Node{int l,r,sum;}T[N<<
2]; inline void dfs1(int p){ siz[p]=1; for(int i=first[p];i;i=e[i].next){ int v=e[i].v; if(v==fa[p])continue; fa[v]=p,dep[v]=dep[p]+1,dfs1(v),siz[p]+=siz[v]; if(siz[v]>siz[hson[p]])hson[p]=v; } } inline void dfs2(int p,int tp){ top[p]=tp,pred[num[p]=++tot]=p; if(!hson[p])return; dfs2
(hson[p],tp); for(int i=first[p];i;i=e[i].next){ int v=e[i].v; if(v!=fa[p]&&v!=hson[p])dfs2(v,v); } } inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;} inline void pushup(int p){T[p].sum=T[lc].sum+T[rc].sum;} inline void build(int p,int l,int r){ T[p].l=l,T[p].r=r; if(l==r)return; build(lc,l,mid),build(rc,mid+1,r); } inline void update(int p,int k){ if(T[p].l==T[p].r){T[p].sum^=1;return;} if(k<=mid)update(lc,k); else update(rc,k); pushup(p); } inline int query(int p,int ql,int qr){ if(!T[p].sum)return -1; if(ql>T[p].r||qr<T[p].l)return -1; if(ql<=T[p].l&&T[p].r<=qr){ if(T[p].l==T[p].r)return pred[T[p].l]; if(T[lc].sum)return query(lc,ql,qr); return query(rc,ql,qr); } if(qr<=mid)return query(lc,ql,qr); if(ql>mid)return query(rc,ql,qr); int ret1=query(lc,ql,qr),ret2=query(rc,ql,qr); if(~ret1)return ret1; if(~ret2)return ret2; return -1; } inline int ask(int x){ int ret=-1; while(top[x]!=1){ int tmp=query(1,num[top[x]],num[x]); if(~tmp)ret=tmp; x=fa[top[x]]; } int tmp=query(1,1,num[x]); return ~tmp?tmp:ret; } int main(){ n=read(),m=read(); for(int i=1;i<n;++i){ int u=read(),v=read(); add(u,v),add(v,u); } dfs1(1),dfs2(1,1),build(1,1,n); while(m--){ int op=read(),v=read(); if(op)printf("%d\n",ask(v)); else update(1,num[v]); } return 0; }