NYOJ 123 士兵杀敌（四）

 1 #include<stdio.h>

 2 int m,n,tree[1000010];

 3 void update(int index,int inc)//注意插线问点与插点问线的区别 

 4 {

 5     while(index>0){

 6         tree[index]+=inc;

 7         index-=index&(-index);

 8     }

 9 }

10 int sum(int index)

11 {

12     int sum=0;

13     while(index<=n){

14         sum+=tree[index];

15         index+=index&(-index);

16     }

17     return sum;

18 }

19 int main()

20 {

21     int from,to,inc,query;

22     char cmd[7];

23     scanf("%d%d",&m,&n);

24     while(m--){

25         scanf("%s",cmd);

26         if(cmd[0]=='A'){

27             scanf("%d%d%d",&from,&to,&inc);

28             update(from-1,-inc);

29             update(to,inc);

30         }else{

31             scanf("%d",&query);

32             printf("%d\n",sum(query));

33         }

34     }

35     return 0;

36 }

 上面是用树状数组写的，一直能AC，重新复习了一下树状数组，以前只用了它的插点问线的功能，今天用了一下它的插线问点的功能，感觉还可以吧，对于线段树与树状数组的利弊，感觉树状数组能完成的功能，线段树都能完成，但反过来就不一定了，还是线段树比较强大，但是有时树状数组在内存与运行效率上却很占优势，比如本题，线段树方法：时间:1828 内存：35384 而树状数组方法：时间：892 内存：4132，果然有区别，先告一段落吧！

今天看了一下士兵杀敌（四）竟然发现还没过，感觉用树状数组能过，用线段树应该也能过啊，用以前提交的代码发现有时AC，有时TL，是用线段树写的，贴上代码：

 1  

 2 #include<stdio.h>

 3  #define N 1000010

 4  struct node{

 5      int l,r;

 6      int inc; 

 7  }tree[3*N];

 8 void build(int l,int r,int i)

 9  {

10      tree[i].l=l;

11      tree[i].r=r;

12      if(l<r){

13          int mid=(l+r)>>1;

14          build(l,mid,i<<1);

15          build(mid+1,r,(i<<1)+1);

16      }

17  }

18 void add(int l,int r,int inc,int i)

19  {

20      if(l==tree[i].l&&tree[i].r==r){//要是l为0的话就不能这样写了，否则RE，死循环 

21          tree[i].inc+=inc;//加的时候单单加到本段，不往下压 

22          return;

23      }

24      int mid=(tree[i].l+tree[i].r)>>1;

25      if(r<=mid) add(l,r,inc,i<<1);

26      else if(l>mid) add(l,r,inc,(i<<1)+1);

27      else{

28          add(l,mid,inc,i<<1);

29          add(mid+1,r,inc,(i<<1)+1);

30      }

31  }

32  inline int query(int k,int i)

33  {

34      if(tree[i].l<tree[i].r){//查询的时候父辈的inc值往下压，直到压到叶子节点为止 

35          if(tree[i].inc){

36              tree[i<<1].inc+=tree[i].inc;

37              tree[(i<<1)+1].inc+=tree[i].inc;

38              tree[i].inc=0;//记得清零 

39          }

40          int mid=(tree[i].l+tree[i].r)>>1;

41          if(k<=mid) return query(k,i<<1);

42          else return query(k,(i<<1)+1);

43      }

44      return tree[i].inc;//返回此叶子节点的值，也即是结果 

45  }

46  int main()

47  {

48      int a,b,inc,n,q;

49      char cmd[6];

50      scanf("%d%d",&q,&n);

51      build(1,n,1);

52      while(q--){

53          scanf("%s",cmd);

54          if(*cmd=='A'){

55              scanf("%d%d%d",&a,&b,&inc);

56              if(!a) a++;//真像讨论区有人说的那样本题测试数据有Bug，a可能为0，坑爹啊！ 

57              add(a,b,inc,1);

58          }else{

59              scanf("%d",&a);

60              printf("%d\n",query(a,1));

61          }

62      }

63      return 0;

64  }