HDU-1166-敌兵布阵(线段树)-CSDN博客

本文链接：https://blog.csdn.net/u013476556/article/details/38443651

敌兵布阵

Problem Description

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果你的程序效率不够高的话，Tidy还是会受到Derek的责骂的.

Input

第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令

Output

对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

Sample Input

  
  
   
   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

Sample Output

初学线段树。。。

/*********************************
*********************************/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>

using namespace std;
const int inf = 202222;

int a[50005];
struct node
{
    int left;
    int right;
    int sum;
} p[inf];

void build (int l,int r ,int i)///建树...
{
    p[i].left = l,p[i].right = r;///根节点
    if (l == r)///节点的左右边界相等  就是根节点...
    {
        p[i].sum = a[l];
        return ;
    }
    int mid = (l + r)/2;
    build (l,mid,i*2);///左子树，节点变为i*2
    build (mid + 1,r,i*2+1);///右子树，节点为i*2+1；
    p[i].sum += p[2*i].sum + p[i*2+1].sum;///在递归返回的过程中逐步累加上去...
}

int Query(int l,int r,int i)///查询...i初始为根节点
{
    int mid = (p[i].left + p[i].right)/2;

    if (p[i].left == l && p[i].right == r)
        return p[i].sum;
    if ( r <= mid)///如果当前节点的右边界比mid小 就说明要查询的节点在左子树
        return Query (l,r,i*2);
    else if (l > mid)///如果当前节点的左边界比mid大，就说明要查询的节点在左子树
        return Query(l,r,i*2+1);
    else ///这种情况就属于跨区间   
        return Query(l,mid,i*2) + Query (mid + 1,r,i*2+1);

}

void Add (int i,int x,int y)///加     从根节点开始查询
{
    int l = p[i].left;
    int r = p[i].right;
    int mid = (l + r) / 2;
    if ( x == l && x == r)///找到 要查询的节点
    {
        p[i].sum += y;
        return ;
    }
    if (x > mid)///在右子树
        Add (i*2 + 1,x,y);
    else///在左子树  
        Add (i*2 ,x,y);
    p[i].sum += y;///在递归返回的过程中逐步累加上去
}
/**加减可以合起来写*/
void Sub(int i,int x,int y)///减    根节点
{
    int l = p[i].left;
    int r = p[i].right;
    int mid = (l + r ) / 2;
    if ( x == l && x == r)
    {
        p[i].sum -= y;
        return ;
    }
    if ( x > mid) 
        Sub (i*2+1,x,y);
    else
        Sub (i*2,x,y);
    p[i].sum -= y;
}

int main ()
{
    int t,n;
    int k = 1;
    scanf ("%d",&t);
    while (t--)
    {
        memset (p,0,sizeof (p));///不加初始化就错。。按理说不应该啊
        printf ("Case %d:\n",k++);

        scanf ("%d",&n);
        for (int i = 1; i <= n; i++)
            scanf ("%d",&a[i]);

        build (1,n,1);

        char s[100];
        while (scanf ("%s",s))
        {
            int x,y;
            if (strcmp ("End",s) == 0)
                break;
            scanf ("%d%d",&x,&y);
            if (strcmp ("Query",s) == 0)
            {
                int fff = Query(x,y,1);///1是从根节点开始
                printf ("%d\n",fff);
            }
            else if (strcmp ("Add",s) == 0)
            {
                Add(1,x,y);///1是从根节点开始
            }
            else if (strcmp ("Sub",s) == 0)
            {
                Sub(1,x,y);
            }
        }

    }
    return 0;
}

也可以将上面的build函数写成两个一个build()只是用来建树一个updata()函数进行数值的更新也能过。。

函数各参数的意义：

void build (int ii,int l,int r) 从ii开始建树边界是l，r；

void updata (int ii,int a,int b)从ii开始进行节点更新将第a个节点的数值更新为b；

void build (int ii,int l,int r)
{
    p[ii].left = l;
    p[ii].right = r;
    p[ii].sum = 0;
    if ( l == r )
        return ;
    else
    {
        int mid = ( l + r ) >> 1;
        build (ii * 2 ,l,mid);
        build (ii * 2 + 1,mid + 1 , r);
    }

}

void updata (int ii,int a,int b)
{
    if ( p[ii].left == a && p[ii].right == a)
    {
        p[ii].sum = b;
    }
    else
    {
        int mid = (p[ii].left + p[ii].right )>>1;
        if ( a <= mid )
            updata (ii * 2,a,b);
        else
            updata (ii * 2 + 1,a,b);
        p[ii].sum = p[ii*2].sum + p[ii*2+1].sum;
    }
}

main函数里要这么写：

 scanf ("%d",&n);
        build (1,1,n);

        for (int i = 1; i <= n; i++)
        {
            scanf ("%d",&x);
            updata (1,i,x);///从1开始 将编号为i 的 节点 更新为x
        }