poj 3321 Apple Tree(线段树)

最新推荐文章于 2022-11-28 09:30:34 发布

原创最新推荐文章于 2022-11-28 09:30:34 发布 · 476 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#ACM #线段树 #poj

线段树同时被 2 个专栏收录

7 篇文章

订阅专栏

poj

7 篇文章

订阅专栏

本文介绍了一种利用树形结构结合线段树解决苹果数量查询与更新的问题。通过深度优先搜索（DFS）为每个节点分配时间戳，转换为区间求和问题，采用线段树高效实现节点状态翻转及查询子树苹果总数。

题目大意：

有一棵树，有n个节点，开始时每个分叉点有1个苹果，现在有m个操作，Q x代表查询x节点及x节点的子节点的苹果总和，C x代表将节点x的状态翻转，有苹果变成没苹果，没苹果变成有苹果。

解题思路：

从节点1开始，进行一次深搜，根据遍历时间为每个节点编号，进入一个新节点时时间加1，离开时，时间标为从上一个节点离开的时间。那么每个节点的进入时间和离开时间就组成了一个区间[l,r]，且当前节点的有无苹果状态存在第l个点上，这个节点及子节点的苹果总和就是[l,r]上的苹果总和。所以问题就变成了一个线段树问题。

注意点：

本题会卡vector，不建议使用。

代码：

#include <iostream>
#include <stdio.h>
#include <vector>

using namespace std;

struct range{
    int l,r;
};

struct node{
    int l,r;
    int sum;
};

struct tNode{
    int to,next;
    tNode(){to = -1;next = -1;}
};

int n,m,nowPos = 0;;
tNode son[100005];
range a[100005];
node tree[400005];

void dfs(int x){
    a[x].l = ++nowPos;//进入新节点时时间加1
    if(son[x].to != -1){
        int now = son[x].to;
        while(now != -1){
            dfs(now);
            now = son[now].next;
        }
    }
    a[x].r = nowPos;//离开时时间为从上一个节点离开的时间
}

void buildTree(int root,int l,int r){
    tree[root].l = l;
    tree[root].r = r;
    tree[root].sum = r - l + 1;
    if(l != r){
        int mid = (l+r)>>1;
        buildTree(2*root,l,mid);
        buildTree(2*root+1,mid+1,r);
    }
}

void updata(int root,int pos){
    if(tree[root].l == pos && tree[root].r == pos){
        tree[root].sum = 1 - tree[root].sum;
        return;
    }
    int mid = (tree[root].l + tree[root].r)>>1;
    if(pos <= mid){
        updata(2*root,pos);
    }
    else{
        updata(2*root+1,pos);
    }
    tree[root].sum = tree[2*root].sum + tree[2*root+1].sum;
}

int query(int root,int l,int r){
    if(tree[root].l == l && tree[root].r == r){
        return tree[root].sum;
    }
    int mid = (tree[root].l + tree[root].r)>>1;
    if(r <= mid){
        return query(2*root,l,r);
    }
    else if(l > mid){
        return query(2*root+1,l,r);
    }
    else{
        return query(2*root,l,mid) + query(2*root+1,mid+1,r);
    }
}

int main()
{
    char s[5];
    int x,y;
    scanf("%d",&n);
    for(int i = 0; i < n-1; i ++){
        scanf("%d%d",&x,&y);
        son[y].next = son[x].to;
        son[x].to = y;
    }
    dfs(1);
    buildTree(1,1,n);
    scanf("%d",&m);
    for(int i = 0; i < m; i ++){
        scanf("%s %d",s,&x);
        if(s[0] == 'C'){
            updata(1,a[x].l);
        }
        else{
            printf("%d\n",query(1,a[x].l,a[x].r));
        }
    }
    return 0;
}