2018.10.06 spoj QTREE3(树链剖分)

版权声明:随意转载哦......但还是请注明出处吧: https://blog.csdn.net/dreaming__ldx/article/details/82950889

传送门
发现把树投射到序列上。
发现每个黑点都只会对其子树产生影响。
再继续想想好像在投射到序列上之后好像就是查询从点 u u 到根路径上深度最浅的黑点。
也就是 l o g log 段线段树上最靠左的黑点。
这个直接在线段树上二分到叶子结点返回下标就行了。
代码:

#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;
}

猜你喜欢

转载自blog.csdn.net/dreaming__ldx/article/details/82950889