数据结构，线段树HDU1166

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef struct CNode
{
int L,R; //区间起点和终点
int key;//兵力
int add;//这是记录增量 
struct CNode * left, * right;//放置左右孩子指针 
}IT;
IT *creat(int l,int r)//建树，用二叉链做，压力。。 
{
   IT *s=(IT *)malloc(sizeof(IT));
   s->L=l;
   s->R=r;
   s->key=0;
   if(l==r)//左右相等，表示只有自己一个节点了 
   {
    s->left=s->right=NULL;
    s->key=0;
    return  s; 
   }
   else {
           s->left=creat(l,(r+l)/2);//没到底的话就继续往下开设节点 
           s->right=creat((r+l)/2+1,r);//
           s->key=s->left->key+s->right->key;//
           return  s;
        }
}
void change(IT *T,int l,int r,int x)
{
     int mid;
     if(T->L==l&&T->R==r)//左右相等，表示找到了 
     {
        T->key+=x*(r-l+1);//找到后将所有下设的增量都加到总和里 
        T->add=x;//保留增量，不往下加，这里可以节省时间 
        return ;
     }
     if(T->add)
     {
       if(T->left)
        T->left->add+=T->add;
       if(T->right)
        T->right->add+=T->add;//这里要加上去并且清0是因为之前没有往下加，这次或查询或更新，都要改成正确的值 
        T->add=0;
     }
      mid=(T->L+T->R)/2;
     if(r<=mid)
     {
      change(T->left,l,r,x);//如果右边区间小于中部，说明在左侧，更新左子树 
     }
     else if(l>mid)
      change(T->right,l,r,x);//同上 
     else {
             change(T->left,l,mid,x),change(T->right,mid+1,r,x);//同上 
          }
     T->key=T->left->key+T->right->key;//这里很重要，如果没这一句，叶子结点以上的根结点都没能得到更新 
}
int sum(IT *T,int l,int r)//求总和的算法跟更新的算法类似 
{
    int mid=(T->L+T->R)/2;
    if(T->L==l&&T->R==r)//
     {
        return T->key;
     }
     if(T->add)
     {
       if(T->left)
        T->left+=T->add;
       if(T->right)
        T->right+=T->add;
        T->add=0;
     }
     if(r<=mid)
     return sum(T->left,l,r);//
     else if(l>mid)
     return sum(T->right,l,r);//
     else {
             return (sum(T->left,l,mid)+sum(T->right,mid+1,r));//之所以要返回，是为了统计总和 
          }
}
int main()
{
    int i,n,l,r,x,y,t,f=1;
    char s[20];
    IT *T;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d",&n);
       T=creat(1,n);
       for(i=1;i<=n;i++)
       {
          scanf("%d",&x);
          change(T,i,i,x);//边读入边更新，这里把更新当成插入了 
       }
       printf("Case %d:\n",f++);
       while(scanf("%s",s),strcmp(s,"End"))
       {
           scanf("%d%d",&x,&y);
           if(strcmp(s,"Add")==0)
             change(T,x,x,y);
           else if(strcmp(s,"Sub")==0)
                   change(T,x,x,-y);
           else printf("%d\n",sum(T,x,y));
       }
    }
    return 0;
}

View Code

 

http://acm.hdu.edu.cn/showproblem.php?pid=1166

搞一下午的二叉链线段树，终于1A。。。。%>_<%

线段树操作在三点

一，建树：

在网上找不到二叉链的代码，难道我奇葩了sro  orz

结构体里保存有本身信息，增量，左右区间，孩子节点等等，在建树过程中，没用到一次函数，表示要启用一个节点，所以要重新申请，在判断函数给的左右

区间是否相等后再进行操作，如果相等，表示找到要找的区间，赋值后返回，否则递归实现建树

二，更新：

在更新过程中，如果找到区间，就把增量，信息都赋予节点，否则将增量传与孩子节点的增量，记得相加，最后判断要找的区间往左往右，递归完成，

其中要注意的事情是每过一个节点，都要把沿途的所有节点的信息进行修改，这里指总和

三，查询：

查询过程其实跟更新过程类似，只是要简单的返回和而已

转载于:https://www.cnblogs.com/huzhenbo113/p/3214495.html