（线段树）敌兵布阵--hdu--1166 （入门）

链接：

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

 

自己第一次在没有看题解AC出来的线段树，写的可能不是太好，再贴个学长的代码，学习一下

发现自己的Update部分写了很多废话，直接用a[r]里的值就好了，我还传了那么多的值，真傻！

 

代码：

#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
#include<string.h>
using namespace std;

#define Lson r<<1
#define Rson r<<1|1

const int N = 1e6+5;

struct SegmentTree
{
    int L, R;
    int sum, e;
    int Mid()
    {
        return (R+L)>>1;
    }
}a[N<<2];

void BuildSegTree(int r, int L, int R)
{
    a[r].L=L, a[r].R=R;
    a[r].e=0;

    if(L==R)
    {
        scanf("%d", &a[r].sum);
        return ;
    }

    BuildSegTree(Lson, L, a[r].Mid());
    BuildSegTree(Rson, a[r].Mid()+1, R);

    a[r].sum = a[Lson].sum + a[Rson].sum;
}

void Update(int r, int L, int R, int i, int e)
{
    a[r].sum += e;

    if(L==R && L==i)
        return ;

    if(i<=a[r].Mid())
        Update(Lson, L, a[r].Mid(), i, e);
    else if(i>a[r].Mid())
        Update(Rson, a[r].Mid()+1, R, i, e);
}

int Query(int r, int L, int R)
{
    if(a[r].L==L && a[r].R==R)
        return a[r].sum;

    if(R<=a[r].Mid())
       return  Query(Lson, L, R);
    else if(L>a[r].Mid())
       return Query(Rson, L, R);
    else
    {
        long long Lsum = Query(Lson, L, a[r].Mid());
        long long Rsum = Query(Rson, a[r].Mid()+1, R);

        return Lsum + Rsum;
    }
}

int main()
{
  int t, k=1;
  scanf("%d", &t);
  while(t--)
  {
      int n;
      scanf("%d", &n);
      BuildSegTree(1, 1, n);

      char s[20];
      int L, R, i, e;

      printf("Case %d:\n", k++);

      while(scanf("%s", s), strcmp(s, "End"))
      {
          if(strcmp(s, "Query")==0)
          {
              scanf("%d%d", &L, &R);
              printf("%d\n", Query(1, L, R));
          }
          else if(strcmp(s, "Add")==0)
          {
              scanf("%d%d", &i, &e);
              Update(1, 1, n, i, e);
          }
          else
          {
              scanf("%d%d", &i, &e);
              Update(1, 1, n, i, -e);
          }
      }
  }
   return 0;
}

 

学长的代码：

#include<stdio.h>
#include<math.h>
#include<string.h>

#define maxn 50005

struct node
{
    int L, R, sum;//左右子树,sum记录区间和
    int Mid(){return (L+R)/2;}
}tree[maxn*4];//为了保险起见一般定义是四倍
int val[maxn];//保存每个阵地原来的人数

void Build(int root, int L, int R)//建树
{
    tree[root].L = L, tree[root].R = R;

    if(L == R)
    {
        tree[root].sum = val[L];
        return ;
    }

    Build(root<<1, L, tree[root].Mid());//<<1 运算符相当于乘上 2，因为是数往左移一位
    Build(root<<1|1, tree[root].Mid()+1, R);//|1， 因为左移后最后一位是0， 所以与1进行|相当于+1

    tree[root].sum = tree[root<<1].sum+tree[root<<1|1].sum;//区间和等于左右区间的和
}
//k表示需要更新的点，e表示需要更新的值
void Insert(int root, int k, int e)
{
    tree[root].sum += e;
    if(tree[root].L == tree[root].R)
        return ;

    if(k <= tree[root].Mid())
        Insert(root<<1, k, e);
    else
        Insert(root<<1|1, k, e);
}
//查询区间LR的和
int Query(int root, int L, int R)
{
    if(tree[root].L == L && tree[root].R == R)
        return tree[root].sum;

    //如果在左子树区间
    if(R <= tree[root].Mid())
        return Query(root<<1, L, R);
    else if(L > tree[root].Mid())//如果在右子树区间
        return Query(root<<1|1, L, R);
    else
    {//在左右子树
        int Lsum = Query(root<<1, L, tree[root].Mid());
        int Rsum = Query(root<<1|1, tree[root].Mid()+1, R);

        return Lsum + Rsum;
    }
}

int main()
{
    int T, t=1;

    scanf("%d", &T);

    while(T--)
    {
        int i, N, x, y;

        scanf("%d", &N);

        for(i=1; i<=N; i++)
            scanf("%d", &val[i]);
        Build(1, 1, N);

        char s[10];

        printf("Case %d:\n", t++);

        while(scanf("%s", s), s[0] != 'E')
        {
            scanf("%d%d", &x, &y);

            if(s[0] == 'A')
                Insert(1, x, y);
            else if(s[0] == 'S')
                Insert(1, x, -y);
            else
            {
                int ans = Query(1, x, y);
                printf("%d\n", ans);
            }
        }
    }

    return 0;
}