SPOJ - QTREE3 Query on a tree again! && 洛谷P4092 [HEOI2016/TJOI2016]树【树链剖分】

SPOJ - QTREE3 Query on a tree again!

题目描述

给出N个点的一棵树（N-1条边），节点有白有黑，初始全为白
有两种操作：
0 i : 改变某点的颜色（原来是黑的变白，原来是白的变黑）
1 v : 询问1到v的路径上的第一个黑点，若无，输出-1

输入格式：

第一行 N，Q，表示N个点和Q个操作
第二行到第N行N-1条无向边
再之后Q行，每行一个操作"0 i" 或者"1 v" (1 ≤ i, v ≤ N).

输出格式：

对每个1 v操作输出结果

说明

For 1/3 of the test cases, N=5000, Q=400000.
For 1/3 of the test cases, N=10000, Q=300000.
For 1/3 of the test cases, N=100000, Q=100000.

---

题目分析：

用点权为0表示白色，点权为1表示黑色
树剖后对每个询问
找到最靠近1号节点的点权和>=1的u到top[u]的链
令ll=num[ top[u] ], rr=num[u]
再这个区间中二分查找num值最小的点权为1的点输出

---

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

int n,m;
struct node{int v,nxt;}E[300010];
int head[200010],tot;
int dep[200010],fa[200010],son[200010],size[200010];
int top[200010],num[500010],pos[500010],cnt;
int sum[500010];

void add(int u,int v)
{
    E[++tot].nxt=head[u];
    E[tot].v=v;
    head[u]=tot;
} 

void dfs1(int u,int pa)
{
    size[u]=1;
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==pa) continue;
        dep[v]=dep[u]+1; fa[v]=u;
        dfs1(v,u);
        size[u]+=size[v];
        if(size[v]>size[son[u]]) son[u]=v;
    }
}

void dfs2(int u,int tp) 
{
    top[u]=tp; num[u]=++cnt; pos[cnt]=u;
    if(son[u]) dfs2(son[u],tp);
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==fa[u]||v==son[u]) continue;
        dfs2(v,v);
    }
}

void update(int u,int s,int t,int p)
{
    if(s==t){sum[p]^=1; return;}//修改颜色直接取反
    int mid=s+t>>1;
    if(u<=mid) update(u,s,mid,p<<1);
    else update(u,mid+1,t,p<<1|1);
    sum[p]=sum[p<<1]+sum[p<<1|1];
}

int get(int ll,int rr,int s,int t,int p)
{
    if(ll<=s&&t<=rr) return sum[p];
    int mid=s+t>>1;
    int ans=0;
    if(ll<=mid) ans+=get(ll,rr,s,mid,p<<1);
    if(rr>mid) ans+=get(ll,rr,mid+1,t,p<<1|1);
    return ans; 
}

int check(int ll,int rr)
{
    if(ll==rr) return ll;
    int mid=ll+rr>>1;
    int tp=get(ll,mid,1,n,1);
    if(tp>=1) return check(ll,mid);//尽量靠近1号点
    else return check(mid+1,rr);
} 

int query(int u)
{
    int ll=0,rr=0;
    while(top[u]!=1)
    {
        int tp=get(num[top[u]],num[u],1,n,1);
        if(tp>=1) ll=num[top[u]],rr=num[u];//找到一个权值和>=1的区间先记录
        u=fa[top[u]];
    }
    int tp=get(1,num[u],1,n,1);//以1号点为链顶的最后一条链
    if(tp>=1) ll=1,rr=num[u];
    
    if(ll==0&&rr==0) return -1;//没有黑点
    else return pos[check(ll,rr)];//二分查找
} 

int main()
{
    n=read();m=read();
    for(int i=1;i<n;++i)
    {
        int u=read(),v=read();
        add(u,v);add(v,u);
    }
    
    dep[1]=1;
    dfs1(1,0);dfs2(1,1);
    
    while(m--)
    {
        int k=read(),u=read();
        if(k==0) update(num[u],1,n,1);
        else if(k==1) printf("%d\n",query(u));
    }
    return 0;
}

---

洛谷P4092 [HEOI2016/TJOI2016]树

题目描述

在 2016 年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树，根为 11，有以下两种操作：
标记操作：对某个结点打上标记。（在最开始，只有结点 1 有标记，其他结点均无标记，而且对于某个结点，可以打多次标记。）
询问操作：询问某个结点最近的一个打了标记的祖先。（这个结点本身也算自己的祖先）
你能帮帮她吗?

说明

30% 的数据，1⩽N,Q⩽1000 ；
70% 的数据，1⩽N,Q⩽10000 ；
100% 的数据，1⩽N,Q⩽100000

---

和上一题查询的路径刚好反过来而已，双倍经验

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

int n,m;
struct node{int v,nxt;}E[300010];
int head[200010],tot;
int dep[200010],fa[200010],son[200010],size[200010];
int top[200010],num[500010],pos[500010],cnt;
int sum[500010];

void add(int u,int v)
{
    E[++tot].nxt=head[u];
    E[tot].v=v;
    head[u]=tot;
} 

void dfs1(int u,int pa)
{
    size[u]=1;
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==pa) continue;
        dep[v]=dep[u]+1;  fa[v]=u;
        dfs1(v,u);
        size[u]+=size[v];
        if(size[v]>size[son[u]]) son[u]=v;
    }
}

void dfs2(int u,int tp)
{
    num[u]=++cnt; pos[cnt]=u; top[u]=tp;
    if(son[u]) dfs2(son[u],tp);
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==fa[u]||v==son[u]) continue;
        dfs2(v,v);
    }
}

void update(int u,int s,int t,int p)
{
    if(s==t){sum[p]=1; return;}
    int mid=s+t>>1;
    if(u<=mid) update(u,s,mid,p<<1);
    else update(u,mid+1,t,p<<1|1);
    sum[p]=sum[p<<1]+sum[p<<1|1];
}

int get(int ll,int rr,int s,int t,int p)
{
    if(ll<=s&&t<=rr) return sum[p];
    int mid=s+t>>1;
    int ans=0;
    if(ll<=mid) ans+=get(ll,rr,s,mid,p<<1);
    if(rr>mid) ans+=get(ll,rr,mid+1,t,p<<1|1);
    return ans;
}

int check(int ll,int rr)
{
    if(ll==rr) return ll;
    int mid=ll+rr>>1;
    int tp=get(mid+1,rr,1,n,1);
    if(tp>=1) return check(mid+1,rr);
    else return check(ll,mid);
}

int query(int u)
{
    while(top[u]!=1)
    {
        int tp=get(num[top[u]],num[u],1,n,1);
        if(tp>=1) return check(num[top[u]],num[u]);
        u=fa[top[u]];
    }
    return check(1,num[u]);
}

int main()
{
    n=read();m=read();
    for(int i=1;i<n;++i)
    {
        int u=read(),v=read();
        add(u,v);add(v,u);
    }
    
    dep[1]=1;
    dfs1(1,0); dfs2(1,1);
    sum[num[1]]=1;
    
    while(m--)
    {
        char ss; scanf("%s",&ss);
        int u=read();
        if(ss=='C') update(num[u],1,n,1);
        else if(ss=='Q') printf("%d\n",pos[query(u)]);
    }
    return 0;
}