树状数组 or 线段树 (HDU 1556)

链接:

树状数组

例题:HDU 1556

  N个气球排成一排,从左到右依次编号为1,2,3....N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽"牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了,你能帮他算出每个气球被涂过几次颜色吗?

输入:

每个测试实例第一行为一个整数N,(N <= 100000).接下来的N行,每行包括2个整数a b(1 <= a <= b <= N)。 
当N = 0,输入结束。

输出:

每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数。

样例:

input:

 

3
1 1
2 2
3 3
3
1 1
1 2
1 3
0
output:
1 1 1
3 2 1
 1 #include <stdio.h>
 2 #include <string.h>
 3 const int MAXN=110000;
 4 int n,c[MAXN];
 5 int lowbit(int x)
 6 //计算2^k
 7 {
 8     x=x&-x;
 9     return x;
10 }
11 void update(int num,int val)
12 //向下查询,num是要更新的子节点,val是要修改的值
13 {
14     while(num>0)
15     {
16         c[num]+=val;
17         num-=lowbit(num);
18     }
19 }
20 int getSum(int num)
21 //向上统计每个区间被染色的次数
22 {
23     int sum=0;
24     while(num<=n)
25     {
26         sum+=c[num];
27         num+=lowbit(num);
28     }
29     return sum;
30 }
31 int main()
32 {
33     int a,b;
34     while(scanf("%d",&n),n)
35     {
36         memset(c,0,sizeof(c));
37         for(int i=0;i<n;i++)
38         {
39             scanf("%d%d",&a,&b);
40             //将b以下区间+1
41             update(b,1);
42             //将a以下区间-1
43             update(a-1,-1);
44         }
45         for(int j=1;j<n;j++)
46         {
47             printf("%d ",getSum(j));
48         }
49         printf("%d\n",getSum(n));
50     }
51     return 0;
52 }

 线段树的写法:

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <algorithm>
 5 using namespace std;
 6 const int maxn=100010;
 7 int n;
 8 struct node{
 9     int l;
10     int r;
11     int laz;
12     int sum;
13 }e[maxn<<2];
14 void build(int l,int r,int cur)
15 {
16     e[cur].l=l;
17     e[cur].r=r;
18     e[cur].sum=0;
19     e[cur].laz=0;
20     if(l==r)
21         return;
22     int mid=(l+r)/2;
23     build(l,mid,cur<<1);
24     build(mid+1,r,cur<<1|1);
25 }
26 void pushdown(int cur)
27 {
28     if(e[cur].laz!=0)
29     {
30         e[cur<<1].laz+=e[cur].laz;
31         e[cur<<1|1].laz+=e[cur].laz;
32         e[cur].laz=0;
33     }
34     return;
35 }
36 void update(int pl,int pr,int cur)
37 {
38     if(pl<=e[cur].l&&e[cur].r<=pr)
39     {
40         e[cur].laz++;
41         return;
42     }
43     int mid=(e[cur].l+e[cur].r)/2;
44     if(pl<=mid)
45         update(pl,pr,cur<<1);
46     if(pr>mid)
47         update(pl,pr,cur<<1|1);
48 }
49 void update1(int cur)
50 {
51     if(e[cur].l==e[cur].r)
52     {
53         return;
54     }
55     pushdown(cur);
56     update1(cur<<1);
57     update1(cur<<1|1);
58 }
59 int query(int pl,int pr,int cur)
60 {
61     if(pl<=e[cur].l&&e[cur].r<=pr)
62     {
63         return e[cur].laz;
64     }
65     int mid=(e[cur].l+e[cur].r)/2;
66     int ans=0;
67     if(pl<=mid)
68         ans+=query(pl,pr,cur<<1);
69     if(pr>mid)
70         ans+=query(pl,pr,cur<<1|1);
71 return ans;
72 }
73 int main()
74 {
75     int a,b;
76     while(~scanf("%d",&n)&&n)
77     {
78         build(1,100000,1);
79         for(int i=0;i<n;i++)
80         {
81             scanf("%d%d",&a,&b);
82             update(a,b,1);
83         }
84         update1(1);
85         for(int i=1;i<n;i++)
86         {
87             printf("%d ",query(i,i,1));
88         }
89         printf("%d\n",query(n,n,1));
90     }
91 return 0;
92 }

 

 

posted @ 2018-07-09 00:24  *starry*  阅读(133)  评论(0编辑  收藏  举报