spoj2798 QTREE3 Query on a tree again!

题目链接：http://www.spoj.com/problems/QTREE3/en/

题目大意：

N个节点的树，边的编号为1~N-1，每条边有一个权值，要求模拟两种操作：
0 x :把x点染成黑色的
1 x :求1到x的路径中遇到的第一个黑色点的编号（一开始都是白色的）

题解：树剖和LCT都行？我这次用了树剖..

维护树链的线段数存一下所管的范围中从左往右出现第一个黑球的编号，再搞搞就好了

记得每次更新第一个黑球的编号..

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 201000

struct node
{
	int x,y,next;
}a[maxn],e[maxn];
int len,first[maxn];
struct trnode
{
	int l,r,lc,rc,c;//la
}tr[maxn];int trlen;
int z,son[maxn],fa[maxn],dep[maxn];
int ys[maxn],yys[maxn],top[maxn],tot[maxn];
int mymax(int x,int y){return (x>y)?x:y;}
void ins(int x,int y)
{
	len++;
	a[len].x=x;a[len].y=y;
	a[len].next=first[x];first[x]=len;
}
void dfs1(int x)
{
	son[x]=0;tot[x]=1;
	for (int k=first[x];k!=-1;k=a[k].next)
	{
		int y=a[k].y;
		if (y!=fa[x])
		{
			dep[y]=dep[x]+1;
			fa[y]=x;
			dfs1(y);
			if (tot[son[x]]<tot[y]) son[x]=y;
			tot[x]+=tot[y];
		}
	}
}
void dfs2(int x,int tp)
{
	ys[x]=++z;yys[z]=x;top[x]=tp;
	if (son[x]!=0) dfs2(son[x],tp);
	for (int k=first[x];k!=-1;k=a[k].next)
	{
		int y=a[k].y;
		if (y!=fa[x] && y!=son[x])
		 dfs2(y,y);
	}
}
void bt(int l,int r)
{
	trlen++;int now=trlen;
	tr[now].l=l;tr[now].r=r;
	tr[now].lc=tr[now].rc=-1;
	tr[now].c=0;//tr[now].la=0;
	if (l<r)
	{
		int mid=(l+r)>>1;
		tr[now].lc=trlen+1;bt(l,mid);
		tr[now].rc=trlen+1;bt(mid+1,r);
	}
}
void change(int now,int x)
{
	if (tr[now].l==tr[now].r) {if (tr[now].c==0) tr[now].c=x;else tr[now].c=0;return;}
	int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)>>1;
	if (x<=mid) change(lc,x);
	else change(rc,x);
	if (tr[lc].c!=0) tr[now].c=tr[lc].c;
	else tr[now].c=tr[rc].c;
}
int findb(int now,int l,int r)
{
	if (tr[now].l==l && tr[now].r==r) return tr[now].c;
	int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)>>1;
	if (r<=mid) return findb(lc,l,r);
	else if (l>mid) return findb(rc,l,r);
	else
	{
		int ls=findb(lc,l,mid);
		return (ls==0)?findb(rc,mid+1,r):ls;
	}
}
int query(int x)
{
	int ans=-1,tx=top[x],ls;
	while (1)
	{
		ls=findb(1,ys[tx],ys[x]);
		if (ls!=0) ans=yys[ls];
		x=fa[tx];tx=top[x];
		if (x==0) break;
	}return ans;
}
int main()
{
	//freopen("qtree3.in","r",stdin);
	//freopen("a.out","w",stdout);
	int n,m,x,y,i;
	scanf("%d%d",&n,&m);
	len=0;memset(first,-1,sizeof(first));
	for (i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		ins(x,y);ins(y,x);
		e[i].x=x;e[i].y=y;
	}e[n].x=0;e[n].y=1;
	ins(0,1);ins(1,0);
	dep[0]=0;fa[0]=0;dfs1(0);
	z=0;dfs2(0,0);
	trlen=0;bt(1,n+1);
	while (m--)
	{
		scanf("%d%d",&x,&y);
		if (x==0) change(1,ys[y]);
		else if (x==1) printf("%d\n",query(y));
	}
	return 0;
}

啊 想要对拍程序的可以找我