zoukankan      html  css  js  c++  java
  • 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
    ",query(y));
    	}
    	return 0;
    }

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

  • 相关阅读:
    人生如此
    微软十七道智力面试题及答案
    【Flink系列十】Flink作业提交过程的调试和诊断
    【Flink系列九】Flink 作业提交遇到的问题记录以及原理
    Jackson ObjectMapper JSON序列化工具使用笔记,由浅入深
    既有设计模式的lambda重构
    观察者模式/Observer
    函数式接口java.util.function
    面向对象世界的七个设计原则
    重构-改善既有代码设计读后灵光
  • 原文地址:https://www.cnblogs.com/Euryale-Rose/p/6527873.html
Copyright © 2011-2022 走看看