敌兵布阵 ---线段树单点修改+区间求和

原创于 2020-02-29 16:20:26 发布 · 162 阅读

0 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了一种基于段式树状数组的数据结构实现方法，该方法主要用于解决区间查询和更新操作的问题。通过实例讲解了如何构建树状结构，进行区间求和、添加和删除操作。适用于需要高效处理大量区间操作的场景。

https://vjudge.net/problem/HDU-1166

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<string>
#include<vector>
#include<stack>
#include<map>
#include<set>
using namespace std;
#define ll long long
const int maxn=50005;
int a[maxn];
int sum;
struct tree
{
    int l,r;
    int dat;
} t[140000];
void build(int p,int l,int r)  ///建树
{
    t[p].l=l;
    t[p].r=r;
    if(l==r)
    {
        t[p].dat=a[l];
        return;
    }
    int mid=(l+r)/2;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    t[p].dat=t[p*2].dat+t[p*2+1].dat;
}

void add(int p,int x,int v) 
{  ///p是当前的点
    t[p].dat+=v; /// 要加v的点的根节点的所有父节点都要加v
    if(t[p].l==t[p].r)
    {
        return;
    }
    int mid=(t[p].l+t[p].r)/2;
    if(x<=mid) add(p*2,x,v);
    else add(p*2+1,x,v);
}

void sup(int p,int x,int v) 
{
    t[p].dat-=v;
    if(t[p].l==t[p].r)
    {
        return;
    }
    int mid=(t[p].l+t[p].r)/2;
    if(x<=mid) sup(p*2,x,v);
    else sup(p*2+1,x,v);
}
int ask(int p,int l,int r) 
{
    if(l<=t[p].l&&r>=t[p].r)
    {
        sum+=t[p].dat;
    }
    else
    {
        int mid=(t[p].l+t[p].r)/2;
        if(r<=mid) ask(p*2,l,r);    ///都在左区间
        else if(l>=mid+1)  ask(p*2+1,l,r);///都在右区间
        else   ///即在左区间又在右区间
        {
            ask(p*2,l,r);
            ask(p*2+1,l,r);
        }
    }
}
int main()
{
    int t;
    int n;
    int cnt=0;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        build(1,1,n);
        printf("Case %d:\n",++cnt);
        char s[10];
        int a,b;
        while(1)
        {
            scanf("%s",s);
            if(s[0]=='E') break;

            if(s[0]=='Q')
            {
                scanf("%d%d",&a,&b);
                sum=0;
                ask(1,a,b);
                printf("%d\n",sum);
            }
            else if(s[0]=='A')
            {
                scanf("%d%d",&a,&b);
                add(1,a,b);
            }
            else if(s[0]=='S')
            {
                scanf("%d%d",&a,&b);
                sup(1,a,b);
            }
        }
    }
}