2018.10.06 spoj QTREE3(樹鏈剖分)
阿新 • • 發佈:2018-12-13
傳送門 發現把樹投射到序列上。 發現每個黑點都只會對其子樹產生影響。 再繼續想想好像在投射到序列上之後好像就是查詢從點到根路徑上深度最淺的黑點。 也就是段線段樹上最靠左的黑點。 這個直接線上段樹上二分到葉子結點返回下標就行了。 程式碼:
#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;
}