数据结构——线段树之一(C++)

源代码:

#include<cstdio>
int m,n,num(0),h[100001];
struct treetype
{
    int left,right,lefts,rights,sum; //本代码中,应用为半开半闭区间。
}i[200020]; //应用了完全二叉树的结点个数公式。
void x1(int t1,int t2) //建树。
{
    int t=++num; //注意,在线段树中,始终遵循左小右大原则。
    i[t].left=t1;
    i[t].right=t2;
    if (t1!=t2-1)
    {
      i[t].lefts=num+1; //利用左小右大原则,进行编号。
      x1(t1,(t1+t2)/2);
      i[t].rights=num+1;
      x1((t1+t2)/2,t2); //注意,此为半开半闭区间。
      i[t].sum=i[i[t].lefts].sum+i[i[t].rights].sum; //利用回溯,进行区间和的更新。
    }
    else
      i[t].sum=h[t1];
}
void x2(int t,int x,int y) //修改。
{
    if (i[t].left==i[t].right-1) //注意,此为半开半闭区间。
      i[t].sum+=y;
    else //已经找到并处理了结点,故不再进行重复且错误的处理了。
    {
      if (x<(i[t].left+i[t].right)/2) //利用二分法,进行结点的修改。
        x2(i[t].lefts,x,y);
      if (x>=(i[t].left+i[t].right)/2) //注意,此为半开半闭区间。
        x2(i[t].rights,x,y);
      i[t].sum=i[i[t].lefts].sum+i[i[t].rights].sum; //利用回溯,进行区间和的更新。
    }
}
int x3(int t,int x,int y) //求和。
{
    if (x<=i[t].left&&y>=i[t].right) //此时,所求区间包含此区间。
      return i[t].sum;
    int ans(0);
    if (x<(i[t].left+i[t].right)/2) //将所求区间分割开来。
      ans+=x3(i[t].lefts,x,y);
    if (y>(i[t].left+i[t].right)/2) //注意,此为半开半闭区间。
      ans+=x3(i[t].rights,x,y);
    return ans;
}
int main()
{
    scanf("%d",&n);
    for (int a=1;a<=n;a++)
      scanf("%d",&h[a]);
    x1(1,n+1); //注意,此为半开半闭区间。
    scanf("%d",&m);
    for (int a=1;a<=m;a++)
    {
      int x,y,z;
      scanf("%d%d%d",&x,&y,&z);
      if (x==1)
        x2(1,y,z);
      else
        printf("%d\n",x3(1,y,z+1)); //注意,此为半开半闭区间。
    }
    return 0;
}
posted @ 2016-02-13 20:08  【機關】  阅读(626)  评论(0编辑  收藏  举报