【线段树】模板-HDU-1166-点改区查

最新推荐文章于 2021-07-19 17:11:09 发布

末树之墨

最新推荐文章于 2021-07-19 17:11:09 发布

阅读量643

点赞数

分类专栏：基础算法题例数据结构文章标签： ACM 数据结构线段树模板

本文链接：https://blog.csdn.net/cryssdut/article/details/38922661

版权

基础算法题例同时被 2 个专栏收录

38 篇文章 0 订阅

订阅专栏

数据结构

4 篇文章 0 订阅

订阅专栏

这篇博客保存了一份线段树的模板，作者强调了其在ACM竞赛中多次应用且效果良好。模板来源于知名选手hh，同时提供了具体的题目实例——HDU-1166「敌兵布阵」。

摘要由CSDN通过智能技术生成

存个线段树模板，已经用过好多次了，好用的

【来自hh大神的模板】

例题链接： HDU-1166-敌兵布阵

#include <cstdio>
#include <iostream>
using namespace std;

#define lson l, m ,rt<<1
#define rson m+1, r, rt<<1|1
const int maxn = 50050;
int sum[maxn<<2];
void PushUP(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void build(int l,int r,int rt)
{
    if(l == r)
    {
        scanf("%d",&sum[rt]);
        return;
    }
    int m = (l+r) >>1;
    build(lson);
    build(rson);
    PushUP(rt);
}

void update(int p,int add,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt] += add;
        return;
    }
    int m = (l+r) >>1 ;
    if(p <= m) update(p,add,lson);
    else update(p,add,rson);
    PushUP(rt);
}

int query(int L,int R,int l,int r,int rt)
{
    if(L<=l && r<=R) return sum[rt];
    int m = (l+r) >>1;
    int ret = 0;
    if(L <= m)ret += query(L, R, lson);
    if(R > m) ret += query(L, R, rson);
    return ret;
}

int main()
{
    freopen("input.txt","r",stdin);
    int T,n,cc=1;
    cin>>T;
    while(T--)
    {
        printf("Case %d:\n",cc);cc++;
        scanf("%d",&n);
        build(1,n,1);
        char op[10];
        while(scanf("%s",op))
        {
            if(op[0] == 'E')break;
            int a,b;
            scanf("%d%d",&a,&b);
            switch(op[0])
            {
                case 'A': update(a,b,1,n,1);break;
                case 'S': update(a,-b,1,n,1);break;
                case 'Q': printf("%d\n",query(a,b,1,n,1));break;
            }
        }
    }
    return 0;
}