数据结构,线段树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

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

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

二,更新:

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

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

三,查询:

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

posted @ 2013-07-25 15:46  执着追求的IT小小鸟  阅读(176)  评论(0编辑  收藏  举报