POJ 3321 Apple Tree

2022-02-02 15:01:38 阅读：142 来源： 互联网

标签：head Apple int dfs 3321 edge POJ 100010 out

题目大意：

题解：
\(dfs\)序的两个数组\(in\)，\(out\)记录某结点进入和出来的时间，则对于一个结点\(u\)，其子树中某一结点\(v\)满足\(in_u < in_v < out_v < out_u\)。
用树状数组存储各时间戳所对应的结点的状况，则\(getSum(out_u) - getSum(in_u - 1)\)所得结果就是以\(u\)为根的子树的状况。
由于每次进入或者出来时间戳都会加一，所以树状数组的大小应为\(2 \times n\)。

#include <iostream>
using namespace std;

struct Edge {
    int v, next;
} edge[100010];
char s;
int x;
int in[100010], out[100010], sum[200010], tot;
int n, m, cnt, head[100010];
bool vis[100010];

void addEdge(int u, int v) {
    edge[++cnt].v = v;
    edge[cnt].next = head[u];
    head[u] = cnt;
}

int lowbit(int x) { return x & (-x); }

void dfs(int x) {
    in[x] = ++tot;
    for (int i = head[x]; i; i = edge[i].next) {
        dfs(edge[i].v);
    }
    out[x] = ++tot;
}

void update(int x, int add) {  // 树状数组更新
    while (x <= 2 * n) {
        sum[x] += add;
        x += lowbit(x);
    }
}

int getSum(int x) {
    int ans = 0;
    while (x > 0) {
        ans += sum[x];
        x -= lowbit(x);
    }
    return ans;
}

int main() {
    ios::sync_with_stdio(false);
    cin >> n;
    for (int i = 1; i < n; ++i) {
        int u, v;
        cin >> u >> v;
        addEdge(u, v);
    }
    dfs(1);
    for (int i = 1; i <= n; ++i) {
        update(in[i], 1);
        vis[i] = 1;
    }
    cin >> m;
    while (m--) {
        cin >> s >> x;
        if (s == 'C') {
            if (vis[x]) {
                update(in[x], -1);
            } else {
                update(in[x], 1);
            }
            vis[x] ^= 1;
        }
        if (s == 'Q') {
            cout << getSum(out[x]) - getSum(in[x] - 1) << endl;
        }
    }
    return 0;
}

标签：head,Apple,int,dfs,3321,edge,POJ,100010,out
来源： https://www.cnblogs.com/IzumiSagiri/p/15860335.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

POJ 3321 Apple Tree