线段树之单点更新求和hdoj1166

逃离我推掉我的手 2022-09-20 15:21 296阅读 0赞

题目:hdoj1166

分析:题意很清晰,就是让你给某个点又增加或者减少x个,然后求某一段有多少个,我是用一个father数组保存叶子节点的编号,然后直接从当前节点开始,更轻到root就ok。

查询的话,从根节点开始,看在左子区间还是右子区间,直接查询到某一段全部在要查询的区间内,求和就ok,很简单。

代码:

  1. #include <cstdio>
  2. #include <string>
  3. #include <iostream>
  4. using namespace std;
  5. const int N = 55000;
  6. struct Node
  7. {
  8. int l,r;
  9. int sum;
  10. };
  11. Node tree[4*N];
  12. int a[N];
  13. int fa[N];
  14. void build(int l,int r,int v)
  15. {
  16. tree[v].l=l;
  17. tree[v].r=r;
  18. if(l==r)
  19. {
  20. fa[l]=v;
  21. tree[v].sum=a[l];
  22. return ;
  23. }
  24. int mid=(l+r)>>1;
  25. build(l,mid,v*2);
  26. build(mid+1,r,v*2+1);
  27. tree[v].sum=tree[v+v].sum+tree[v+v+1].sum;
  28. }
  29. void update(int t,int p)
  30. {
  31. tree[t].sum+=p;
  32. if(t==1)
  33. return ;
  34. update(t/2,p);
  35. }
  36. int queue(int a,int b,int v)
  37. {
  38. if(tree[v].l==a&&tree[v].r==b)
  39. {
  40. return tree[v].sum;
  41. }
  42. int mid=(tree[v].l+tree[v].r)>>1;
  43. if(b<=mid) return queue(a,b,v+v);
  44. else if(a>mid) return queue(a,b,v+v+1);
  45. else return queue(a,mid,v*2)+queue(mid+1,b,v*2+1);
  46. }
  47. int main()
  48. {
  49. int T,n;
  50. char str[10];
  51. int x,y;
  52. scanf("%d",&T);
  53. for(int cas=1; cas<=T; cas++)
  54. {
  55. scanf("%d",&n);
  56. for(int i=1; i<=n; i++)
  57. scanf("%d",&a[i]);
  58. build(1,N,1);
  59. printf("Case %d:\n",cas);
  60. while(scanf("%s",str)>0)
  61. {
  62. if(str[0]=='A')
  63. {
  64. scanf("%d %d",&x,&y);
  65. update(fa[x],y);
  66. }
  67. else if(str[0]=='S')
  68. {
  69. scanf("%d %d",&x,&y);
  70. update(fa[x],-y);
  71. }
  72. else if(str[0]=='Q')
  73. {
  74. scanf("%d %d",&x,&y);
  75. int ans=queue(x,y,1);
  76. printf("%d\n",ans);
  77. }
  78. else
  79. {
  80. break;
  81. }
  82. }
  83. }
  84. }

发表评论

表情:
评论列表 (有 0 条评论,296人围观)

还没有评论,来说两句吧...

相关阅读