POJ 3321 Apple Tree 【树状数组+建树】

题目链接:http://poj.org/problem?id=3321

Apple Tree

Time Limit: 2000MS

Memory Limit: 65536K

Total Submissions: 34812

Accepted: 10469

Description

There is an apple tree outside of kaka‘s house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very much, so he has been carefully nurturing the big apple tree.

The tree has N forks which are connected by branches. Kaka numbers the forks by 1 to N and the root is always numbered by 1. Apples will grow on the forks and two apple won‘t grow on the same fork. kaka wants to know how many apples are there in a sub-tree, for his study of the produce ability of the apple tree.

The trouble is that a new apple may grow on an empty fork some time and kaka may pick an apple from the tree for his dessert. Can you help kaka?

Input

The first line contains an integer N (N ≤ 100,000) , which is the number of the forks in the tree.
The following N - 1 lines each contain two integers u and v, which means fork u and fork v are connected by a branch.
The next line contains an integer M (M ≤ 100,000).
The following M lines each contain a message which is either
"x" which means the existence of the apple on fork x has been changed. i.e. if there is an apple on the fork, then Kaka pick it; otherwise a new apple has grown on the empty fork.
or
"x" which means an inquiry for the number of apples in the sub-tree above the fork x, including the apple (if exists) on the fork x
Note the tree is full of apples at the beginning

Output

For every inquiry, output the correspond answer per line.

Sample Input

3
1 2
1 3
3
Q 1
C 2
Q 1

Sample Output

3
2

Source

POJ Monthly--2007.08.05, Huang, Jinsong

题意概括:

给一个 N 阶无向图,要求把这个无向图转换为一颗以 1 为根的树后进行两种操作:

C x :x结点上如果有苹果就摘掉,如果没有苹果就长出一个新的来

Q x:   查询 x 结点与它的所有后代分支一共有几个苹果

解题思路:

因为原本苹果之间的关系我们不好处理,所以就给他们这些结点重新编号。

并且为了方便子树的苹果数求和(即区间查询)我们在给结点增加一个左值一个右值表示当前结点管辖的范围。

如何建树?

根据题目一开始给的边与边的关系我们可以通过 dfs 进行建树。

L[ x ] (X的左值): X结点的新编号就是该结点的左值,表示它管辖范围的下限

R[ x ] (X的右值):而子树的编号最大值为该结点的右值,表示它管辖范围的上限。

至于dfs顺序,如下

举个栗子:

N = 6;

1 - 2

1 - 6

2 - 5

2 - 3

6 - 4

到这里感觉有一点点线段树的味道了。。。

但这里要用树状数组对这些区间进行维护和查询。

例如:我要 Q 2;

由上面建好的新树我们可以知道 结点2这颗子树包含了结点2、结点5、结点3,而结点2所管辖的区间【2,4】刚好把它们包括进去了。

如果我们要求 这可子树的苹果数量,其实就是用树状数组求区间【2,4】的苹果总数

也就是右值的前缀和减去左值的前缀和(但不包括左值,因为结点本身也要算进去)即:ans = sum( R[ 2 ] ) - sum( L[ 2 ] - 1);

而C操作其实就是树状数组的单点更新而已。

注意事项:

一开始用 stl 的 vector 存无向图,虽然过了但效率不咋地。想想好久没用静态邻接表了,换了一下,快了好多。

AC code:

  1 ///建树+树状数组(stl)8080k 1047ms
  2 /*
  3 #include <cstdio>
  4 #include <iostream>
  5 #include <cstring>
  6 #include <vector>
  7 #include <algorithm>
  8 #define INF 0x3f3f3f3f
  9 using namespace std;
 10 typedef vector<int>ve;
 11
 12 const int MAXN  = 1e5+10;
 13 int t[MAXN], l[MAXN], r[MAXN];
 14 bool vis[MAXN];
 15 int N, M, cnt;
 16
 17 vector<ve>node(MAXN);
 18
 19 int lowbit(int x)
 20 {
 21     return x&(-x);
 22 }
 23 void add(int x, int value)
 24 {
 25     for(int i = x; i <= N; i+=lowbit(i))
 26         t[i]+=value;
 27 }
 28 int sum(int x)
 29 {
 30     int res = 0;
 31     for(int i = x; i > 0; i-=lowbit(i))
 32         res+=t[i];
 33     return res;
 34 }
 35 void dfs(int k)
 36 {
 37     l[k] = cnt;
 38     for(int i = 0; i < node[k].size(); i++)
 39     {
 40         cnt++;
 41         dfs(node[k][i]);
 42     }
 43     r[k] = cnt;
 44     return;
 45 }
 46 int main()
 47 {
 48     scanf("%d", &N);
 49     int a, b;
 50     for(int i = 1; i <= N; i++)
 51     {
 52         vis[i] = 1;
 53         add(i, 1);
 54     }
 55     for(int i = 1; i < N; i++)
 56     {
 57         scanf("%d%d", &a, &b);
 58         node[a].push_back(b);
 59     }
 60     cnt = 1;
 61     dfs(1);
 62     scanf("%d", &M);
 63     while(M--)
 64     {
 65         char com[3];
 66         int px;
 67         scanf("%s", &com);
 68         if(com[0] == ‘C‘)
 69         {
 70             scanf("%d", &px);
 71             if(vis[px]) add(l[px], -1);
 72             else add(l[px], 1);
 73             vis[px] = !vis[px];
 74         }
 75         else
 76         {
 77             scanf("%d", &px);
 78             int ans = sum(r[px]) - sum(l[px]-1);
 79             printf("%d\n", ans);
 80         }
 81     }
 82     return 0;
 83 }
 84 */
 85
 86 ///建树+树状数组(静态连接表) 4940k 454ms
 87 #include <cstdio>
 88 #include <iostream>
 89 #include <cstring>
 90 #include <vector>
 91 #include <algorithm>
 92 #define INF 0x3f3f3f3f
 93 using namespace std;
 94
 95 const int MAXN = 1e5+10;
 96 const int MAXM = 1e5+10;
 97
 98 struct node
 99 {
100     int to, next;
101 }edge[MAXM<<1];
102
103 int head[MAXN];
104 bool vis[MAXN];
105 int t[MAXN], l[MAXN], r[MAXN];
106 int N, M, cnt, key;
107
108 int lowbit(int x)
109 {
110     return x&(-x);
111 }
112 void add(int x, int value)
113 {
114     for(int i = x; i <= N; i+=lowbit(i))
115         t[i]+=value;
116 }
117 int sum(int x)
118 {
119     int res = 0;
120     for(int i = x; i > 0; i-=lowbit(i))
121         res+=t[i];
122     return res;
123 }
124
125 void dfs(int k)
126 {
127     l[k] = key;
128     for(int i = head[k]; i != -1; i = edge[i].next)
129     {
130         if(l[edge[i].to]) continue;
131         key++;
132         dfs(edge[i].to);
133     }
134     r[k] = key;
135 }
136
137 void init()
138 {
139     memset(head, -1, sizeof(head));
140     cnt = 1;
141 }
142
143 void add_e(int u, int v)
144 {
145     edge[cnt].to = v;
146     edge[cnt].next = head[u];
147     head[u] = cnt++;
148 }
149
150 int main()
151 {
152     scanf("%d", &N);
153     init();
154     int a, b;
155     for(int i = 1; i <= N; i++)
156     {
157         vis[i] = 1;             //每个节点一开始都有苹果
158         add(i, 1);              //初始化树状数组
159     }
160     for(int i = 1; i < N; i++)
161     {
162         scanf("%d%d", &a, &b); ///建无向图
163         add_e(a, b);
164         add_e(b, a);
165     }
166     key = 1; dfs(1);           //建树
167     scanf("%d", &M);
168     int px;
169     char com[3];
170     while(M--)
171     {
172         scanf("%s", &com);
173         if(com[0] == ‘C‘)
174         {
175             scanf("%d", &px);
176             if(vis[px]) add(l[px], -1);
177             else add(l[px], 1);
178             vis[px] = !vis[px];
179         }
180         else
181         {
182             scanf("%d", &px);
183             printf("%d\n", sum(r[px])-sum(l[px]-1));
184         }
185     }
186     return 0;
187 }

原文地址:https://www.cnblogs.com/ymzjj/p/9471893.html

时间: 2024-10-12 19:18:19

POJ 3321 Apple Tree 【树状数组+建树】的相关文章

POJ 3321 Apple Tree (树状数组)

Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 21191   Accepted: 6436 Description There is an apple tree outside of kaka's house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very much, so he has been

E - Apple Tree(树状数组+DFS序)

There is an apple tree outside of kaka's house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very much, so he has been carefully nurturing the big apple tree. The tree has N forks which are connected by branches. Kaka numbers

POJ3321 Apple Tree(树状数组)

先做一次dfs求得每个节点为根的子树在树状数组中编号的起始值和结束值,再树状数组做区间查询 与单点更新. #include<cstdio>#include<iostream>#include<cstdlib>#include<cstring>#include<string>#include<algorithm>#include<map>#include<queue>#include<vector>#

poj 3321:Apple Tree(树状数组,提高题)

Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 18623   Accepted: 5629 Description There is an apple tree outside of kaka's house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very much, so he has been

【树状数组】POJ 3321 Apple Tree

/** * @author johnsondu * @time 2015.8.25 20:04 * @problem POJ 3321 Apple Tree * @type Binary Index Tree * @description 从根节点开始,dfs遍历树,先访问的节点 * 记为beg, 从当前结点遍历访问到的最后的 * 一个节点,记为end.然后按照树状数组的 * 方法进行求解. * @url http://poj.org/problem?id=3321 */ #include <i

POJ - 3321 Apple Tree (线段树 + 建树 + 思维转换)

POJ - 3321 Apple Tree Time Limit: 2000MS   Memory Limit: 65536KB   64bit IO Format: %I64d & %I64u Submit Status Description There is an apple tree outside of kaka's house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very muc

POJ 3321 Apple Tree (dfs+线段树)

题目大意: 修改树上的节点,然后求子树的和. 思路分析: dfs 重新编号,烂大街了... #include <cstdio> #include <iostream> #include <cstring> #include <algorithm> #define maxn 100005 #define lson num<<1,s,mid #define rson num<<1|1,mid+1,e using namespace std

HDU 3333 Turing Tree 树状数组 离线查询

题意: 给你一个数列,然后有n个查询,问你给定区间中不同数字的和是多少. 思路还是比较难想的,起码对于蒟蒻我来说. 将区间按照先右端点,后左端点从小到大排序之后,对于每个查询,我只要维护每个数字出现的最后一次就可以了(这个结论稍微想一下就可以证明是正确的). 然后就是简单的点更新,区间求和问题了- #include <cstdio> #include <cstring> #include <iostream> #include <map> #include

hdu 1541/poj 2352:Stars(树状数组,经典题)

Stars Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 4052    Accepted Submission(s): 1592 Problem Description Astronomers often examine star maps where stars are represented by points on a plan

POJ 2892 Tunnel Warfare (树状数组+二分)

题目大意: 三个操作 D pos  将pos位置摧毁,让它和周围不相连. Q pos 问和pos 相连的有多少个村庄. R 修复最近摧毁的村庄. 思路分析: 树状数组记录这个区间有多少个1. 如果  [s-e] 有e-s+1个1 的话.那么这个区间是相连的. 这样的话,我们就可以用二分的办法求出与某个位置最大相连的数量. 还有这里二分 while(l<=r) { if(满足) { ans=mid; l=mid+1; } else r=mid-1; } #include <cstdio>