简单线段树–hdu1166

简单线段树–hdu1166

---

这题是一道非常基础的线段树题，最近开始集训打算把线段树重新捡起来写一次，结果发现写的简直差强人意，很多地方调试得非常辛苦，最后还参考了别人的线段树版，但是这次写线段树算是再一次加强了对线段树的理解，由于这个线段树还是比较裸的，所以直接套版都能过。

相对是自己第一次敲线段树版，所以注解比较多，相对初学者来说比较清晰。希望以后在能够对更多的算法有自己的理解=-=

---

下面附上hdu1166 AC代码

---

hdu暂时可以用

hdu当时不可用

---

/****************
create by QuanQ
create time 2016-9-1
example hdu 1166
algorithm 线段树基本单点更新和区间查询
*****************/
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;

#define maxn 500005

struct SegTree{   //线段树
    int l,r,v;    //lr左右孩子,v值
}tree[maxn << 2];

/*创建线段树*/
void build_sTree(int root,int left,int right){
    tree[root].l = left;
    tree[root].r = right;
    tree[root].v = 0;    //总和,初始化为0
    if(left < right){
        int mid = (left + right) >> 1;
        build_sTree(root << 1,left,mid);          //创建左孩子
        build_sTree(root << 1 | 1,mid + 1,right); //创建右孩子
        /*
            注意这里用到的左右移运算符和或运算
            二进制右移1位相当于乘二,那么再或1的话就相当于+1,这里只是为了简化代码
        */
    }
}

/*加入值/更新值*/
void insert_sTree(int value,int root,int need){   //有些代码也叫update,更新值
     /*传入参数：value增加的值,root当前根部--通常传入时为1,need需要更改的key*/
     if(tree[root].l == tree[root].r){
        tree[root].v += value;
        return;
     }
     int mid = (tree[root].l + tree[root].r) >> 1;
     if(need <= mid)
        insert_sTree(value,root << 1,need);
     else
        insert_sTree(value,root << 1 | 1,need);
     //最后出栈的时候更新数据
     tree[root].v = tree[root << 1].v + tree[root << 1 | 1].v;
}

/*查找--区间求和*/
int search_sTree(int root,int l,int r){
    /*传入参数：root当前根部--通常传入时为1,l,r,为所需要从l开始到r的所有数据的区间段,l一定要小于r*/
    int mid = (tree[root].l + tree[root].r) >> 1;
    if(l == tree[root].l && r == tree[root].r)   //如果找到了相应的位置,直接返回值
        return tree[root].v;
    if(r <= mid)                  //如果当前搜索的节点范围比正常范围大,则向左移动
        return search_sTree(root << 1,l,r);
    else if(l > mid)                  //如果当前搜索的节点范围比正常范围小,则向右移动
        return search_sTree(root << 1 | 1,l,r);
    else                                         //如果被当前区间分割,则分开两部分计算
        return search_sTree(root << 1,l,mid) + search_sTree(root << 1 | 1,mid + 1,r);
}

int main(){
    int T,n,value,a,b;
    char ch[100];
    scanf("%d",&T);
    for(int t = 1;t <= T;t++){
        scanf("%d",&n);
        printf("Case %d:\n",t);
        build_sTree(1,1,n);
        for(int i = 1;i <= n;i++){
            scanf("%d",&value);
            insert_sTree(value,1,i);
        }
        while(scanf("%s",ch) && strcmp(ch,"End")){
            scanf("%d %d",&a,&b);
            if(strcmp(ch,"Query") == 0)
                printf("%d\n",search_sTree(1,a,b));
            else if(strcmp(ch,"Add") == 0)
                insert_sTree(b,1,a);
            else if(strcmp(ch,"Sub") == 0)
                insert_sTree(-b,1,a);
        }
    }
}

/*
1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End
*/