spoj-QTREE3 LCT

37 篇文章 0 订阅
8 篇文章 0 订阅

题目大意:一棵树分为黑点和白点,可以改变一个点的颜色,问从1到v的路径上第一个黑点是哪个点

LCT做法:把黑点col记作1,sum[x]记录x点子树右多少个黑点,我们可以用LCT把1到x路径上所有的点组成一个splay,然后在splay上找关键字最小的点就可以了,不难发现这个题就算找路径上第k个点也是很容易实现的。


对于树链剖分的做法也很容易理解,就是剖分不停在向上爬的时候类似于二分查找去寻找第一个出现的黑点在哪里,col和sum数组就变成了线段树的端点修改。

两种方法的复杂度都是nlognlogn

#include <bits/stdc++.h>
using namespace std;
#define maxn 100010
inline int getint()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,Q;
int next[maxn],tr[maxn][2],fa[maxn],col[maxn],sum[maxn];
bool rev[maxn];
vector<int>g[maxn];
bool isroot(int rt)
{
    return tr[fa[rt]][0]!=rt && tr[fa[rt]][1]!=rt;
}
void pushup(int x)
{
    sum[x]=sum[tr[x][0]]+sum[tr[x][1]]+col[x];
}
void pushdown(int x)
{
    if(!isroot(x))
        pushdown(fa[x]);
    if(rev[x])
    {
        rev[x]^=1;rev[tr[x][1]]^=1;rev[tr[x][0]]^=1;
        swap(tr[x][0],tr[x][1]);
    }
}
void rotate(int &x)
{
    int l,r,y,z;
    y=fa[x];z=fa[y];
    if(tr[y][0]==x) l=0;
    else l=1; r=l^1;
    if(!isroot(y))
    {
        if(tr[z][0]==y) tr[z][0]=x;
        else tr[z][1]=x;
    }
    fa[x]=z;fa[y]=x;fa[tr[x][r]]=y;
    tr[y][l]=tr[x][r];tr[x][r]=y;
    pushup(y);pushup(x);
}
void splay(int x)
{
    pushdown(x);
    int y,z;
    while(!isroot(x))
    {
        y=fa[x];z=fa[y];
        if(!isroot(y))
        {
            if((tr[y][0]==x)^(tr[z][0]==y)) rotate(x);
            else rotate(y);
        }
        rotate(x);
    }
}
void access(int x)
{
    int t=0;
    while(x)
    {
        splay(x);
        tr[x][1]=t;
        pushup(x);
        t=x;x=fa[x];
    }
}
void rever(int x)
{
    access(x);splay(x);rev[x]^=1;
}
void link(int x,int y)
{
    rever(x);fa[x]=y;splay(x);
}
void cut(int x,int y)
{
    rever(x);access(y);splay(y);tr[y][0]=fa[x]=0;
}
void dfs(int now,int u)
{
    for(int i=0;i<g[now].size();i++)
    {
        int v=g[now][i];
        if(v==u) continue;
        fa[v]=now;
        dfs(v,now);
    }
}
void update(int rt)
{
    splay(rt);
    col[rt]^=1;
    pushup(rt);
}
void makeroot(int x)
{
    access(x);splay(x);rev[x]^=1;
}
int query(int x)
{
    makeroot(1);
    access(x);splay(x);
    if(sum[x]==0)return -1;
    while(x)
    {
        if(sum[ tr[x][0] ]==0 && col[x]==1 ) return x;
        else if(sum[tr[x][0]]) x=tr[x][0];
        else x=tr[x][1];
    }
}
void init()
{
    for(int i=1;i<=n;i++) g[i].clear();
    memset(col,0,sizeof(col));
    memset(fa,0,sizeof(fa));
    memset(sum,0,sizeof(sum));
    memset(rev,0,sizeof(rev));
    memset(tr,0,sizeof(tr));
}
int main()
{
    while(scanf("%d%d",&n,&Q)!=EOF)
    {
        init();
        int a,b;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        dfs(1,0);
        while(Q--)
        {
            scanf("%d%d",&a,&b);
            if(a==0) update(b);
            else printf("%d\n",query(b));
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值