Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 8984 Accepted Submission(s): 4594
Update(a,1);
Update(b+1,-1);
注意更新a的值就相当于更新区间[a,n]的值(n为限界/因为a要更新往后的所有区间的值),那么更新b+1的值就是更新[b+1,n](更新b往后所有区间的值)。
所以将区间[a,n]值+1,再将无关区间[b+1,n]的值-1,最后就是将区间[a,b]的值+1。
下面是第二类树状数组的实现的两种方式:
代码一(向上更新,向下求和):
1 #include <iostream>
2 using namespace std;
3
4 int n; //限界
5 int c[100010];
6
7 int lowbit(int x)
8 {
9 return x&-x;
10 }
11
12 void Update(int x,int v) //更新区间[x,n]的值
13 {
14 while(x<=n){
15 c[x]+=v;
16 x+=lowbit(x);
17 }
18 }
19
20 int Sum(int x) //求x之前的和
21 {
22 int sum = 0;
23 while(x>0){
24 sum+=c[x];
25 x-=lowbit(x);
26 }
27 return sum;
28 }
29
30 int main()
31 {
32 int a,b,i;
33 while(scanf("%d",&n)!=EOF){
34 memset(c,0,sizeof(c));
35 if(n==0) break;
36 for(i=0;i<n;i++){
37 scanf("%d%d",&a,&b);
38 Update(a,1); //将区间[a,n]的值+1,再将无关区间[b+1,n]的值-1,即将区间[a,b]的值+1
39 Update(b+1,-1);
40 }
41 for(i=1;i<=n;i++){
42 printf("%d",Sum(i)); //求单个的点直接求sum
43 if(i<n)
44 printf(" ");
45 }
46 printf("\n");
47 }
48 return 0;
49 }
代码二(向下更新,向上求和):
1 #include <iostream>
2 using namespace std;
3
4 int n; //限界
5 int c[100010];
6
7 int lowbit(int x)
8 {
9 return x&-x;
10 }
11
12 void Update(int x,int v) //更新区间[x,n]的值
13 {
14 while(x>0){
15 c[x]+=v;
16 x-=lowbit(x); //改动,+变成了-,向下更新
17 }
18 }
19
20 int Sum(int x) //求x之前的和
21 {
22 int sum = 0;
23 while(x<=n){
24 sum+=c[x];
25 x+=lowbit(x); //改动,-变成了+,向上求和
26 }
27 return sum;
28 }
29
30 int main()
31 {
32 int a,b,i;
33 while(scanf("%d",&n)!=EOF){
34 memset(c,0,sizeof(c));
35 if(n==0) break;
36 for(i=0;i<n;i++){
37 scanf("%d%d",&a,&b);
38 Update(a-1,-1); //将区间[1,a-1]的值-1,再将无关区间[1,b]的值+1,即将区间[a,b]的值+1
39 Update(b,1);
40 }
41 for(i=1;i<=n;i++){
42 printf("%d",Sum(i)); //求单个的点直接求sum
43 if(i<n)
44 printf(" ");
45 }
46 printf("\n");
47 }
48 return 0;
49 }
Freecode : www.cnblogs.com/yym2013
hdu 1556:Color the ball(第二类树状数组 —— 区间更新,点求和),布布扣,bubuko.com
hdu 1556:Color the ball(第二类树状数组 —— 区间更新,点求和)
原文:http://www.cnblogs.com/yym2013/p/3873835.html