HYSBZ 4551树

那个就是,一开始要dfs一遍,把那个给每个点的深度,要记录一下,然后就是来像普通的线段树一样建树,更新,查找,就这样就行了。这会写的是左闭右开。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;

const int MaxN = 1e5;
int n , m ;
int all , last[MaxN + 5] , pre[2*MaxN + 5], other[2*MaxN + 5];
int l[MaxN + 5] , r[MaxN + 5] ,tot , dep[MaxN + 5];

int Judge(int x , int y)
{
    if(x == 0) return y;
    if(y == 0) return x;
    if(dep[x] > dep[y]) return x;
    return y;
}
struct SegMentTree{
    int tot;
    struct node{
        int left , right , flag;
    } tree[MaxN << 2 + 5];
    void Buildtree(int x , int l , int r)
    {
        tree[x].left = l,tree[x].right = r,tree[x].flag = 0;
        if(l == r - 1) return ;
        int mid = (l + r) >> 1;
        Buildtree(x << 1 , l , mid);
        Buildtree(x << 1 | 1 , mid , r);
    }
    void Init(int n)
    {
        tot = 1 ;
        while(tot <= n ) tot <<= 1;
        Buildtree(1 , 1 , tot + 1);
    }
    void Push_Down(int x)
    {
        if(tree[x].flag)
        {
            tree[x << 1].flag = Judge(tree[x << 1].flag , tree[x ].flag);
            tree[x << 1 | 1].flag = Judge(tree[x << 1 | 1].flag , tree[x ].flag);
            tree[x].flag = 0;
        }
    }
    void Add_Seg(int x , int l , int r , int t)
    {
        if(tree[x].left == l && tree[x].right == r)
        {
            tree[x].flag = Judge(tree[x].flag , t);
            return ;
        }
        Push_Down(x);
        int mid = (tree[x].left + tree[x].right) >> 1;
        if(r <= mid) Add_Seg(x << 1 , l , r , t);
        else if(l >= mid) Add_Seg(x << 1 | 1 , l , r , t);
        else Add_Seg(x << 1 , l , mid , t) , Add_Seg(x << 1 | 1 , mid , r , t);
    }

    int Ask(int x , int p)
    {
        if(tree[x].left == tree[x].right - 1) return tree[x].flag;
        Push_Down(x);
        int mid = (tree[x].left + tree[x].right) >> 1;
        if(p < mid) return Ask(x << 1 ,p);
        else return Ask(x << 1 | 1 , p);
    }
}P;

void Build(int x , int y)
{
    pre[++all] = last[x];//pre上一条边
    last[x] = all;//x的最后一条边
    other[all] = y; //all这条边的另一个点
}

void Dfs(int x, int y)//y father x son
{
    int ed , dr;
    l[x] = ++tot;
    dep[x] = dep[y] + 1;//深度
    ed = last[x];//当前边
    while(ed != -1){//到头了
        dr = other[ed];
        if(dr != y) Dfs(dr , x);//别让他回去了
        ed = pre[ed];//这条边的上一条边
    }
    r[x] = tot;
}

int main()
{
    while(~scanf("%d%d",&n , &m)){
        all = -1;
        memset(last , - 1 , sizeof(last));
        for(int i = 1 ; i <= n - 1 ; i++){
            int u , v;
            scanf("%d%d", &u , &v);
            Build(u , v);
            Build(v , u);
        }
        tot = 0 , dep[0] = 0;
        Dfs(1 , 0);
        P.Init(n);
        P.Add_Seg(1 , l[1] , r[1] + 1 , 1);
        for(int i = 1 ; i <= m ; i++){
            char c ;
            int x ;
            scanf(" %c",&c);
            scanf("%d",&x);
            if(c == 'C') P.Add_Seg(1 , l[x] , r[x] + 1 , x);//左闭右开
            else printf("%d\n",P.Ask(1, l[x]));
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值