线段树模板

最新推荐文章于 2025-08-13 18:01:09 发布

Cc_Sonia

最新推荐文章于 2025-08-13 18:01:09 发布

阅读量217

点赞数

CC 4.0 BY-SA版权

分类专栏： # 线段树、树状数组文章标签： ACM 算法

本文链接：https://blog.youkuaiyun.com/Cc_Sonia/article/details/82468277

线段树、树状数组专栏收录该内容

8 篇文章

订阅专栏

本文详细解析了线段树的数据结构及其在区间查询和修改操作中的应用，通过具体实例介绍了如何构建、更新和查询线段树，适用于算法竞赛和数据结构学习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

主要参考博客Orz：http://www.cnblogs.com/TheRoadToTheGold/p/6254255.html

通过刷HDU1166和POJ3468，弄出来一份比较合适的模板，记录下来2333

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<queue>
using namespace std;
#define ll long long
typedef pair<int,int>pp;
#define mkp make_pair
#define pb push_back
const int INF=0x3f3f3f3f;
const ll MOD=1e9+7ll;

const int MAX=1e5+5;
int n,q,ans;
struct node
{
    int l,r,w,f; //l，r分别表示区间左右端点，w表示区间和
}tree[4*MAX];

void push_up(int k) //向上更新
{
    tree[k].w=tree[k<<1].w+tree[k<<1|1].w; //注意各操作随题目变化
}
void build(int k,int l,int r) //建树
{
    tree[k].l=l,tree[k].r=r;
    tree[k].f=0;
    if(tree[k].l==tree[k].r)
    {   scanf("%d",&tree[k].w);   return; }
    int m=(tree[k].l+tree[k].r)>>1;
    build(k<<1,l,m);//左孩子结点
    build(k<<1|1,m+1,r);//右孩子结点
    push_up(k);
}
void down(int k) //懒标记下传
{
    tree[k<<1].f+=tree[k].f;
    tree[k<<1|1].f+=tree[k].f;
    tree[k<<1].w+=tree[k].f*(tree[k<<1].r-tree[k<<1].l+1);
    tree[k<<1|1].w+=tree[k].f*(tree[k<<1|1].r-tree[k<<1|1].l+1);
    tree[k].f=0;
}

void ask_point(int k,int x) //单点查询,类似于二分查询
{
    if(tree[k].l==tree[k].r)
    {   ans=tree[k].w;   return;  }
    if(tree[k].f)  down(k);
    int m=(tree[k].l+tree[k].r)>>1;
    if(x<=m)  ask_point(k<<1,x);
    else  ask_point(k<<1|1,x);
}
void change_point(int k,int x,int y) //单点修改
{
    if(tree[k].l==tree[k].r)
    {   tree[k].w+=y;   return;  }
    if(tree[k].f)  down(k);
    int m=(tree[k].l+tree[k].r)>>1;
    if(x<=m)  change_point(k<<1,x,y);
    else  change_point(k<<1|1,x,y);
    push_up(k);
}
void ask_interval(int k,int a,int b) //区间查询
{
    if(tree[k].l>=a&&tree[k].r<=b)
    {   ans+=tree[k].w;   return;  }
    if(tree[k].f)  down(k);
    int m=(tree[k].l+tree[k].r)>>1;
    if(a<=m)  ask_interval(k<<1,a,b);
    if(b>m)  ask_interval(k<<1|1,a,b);
}
void change_interval(int k,int a,int b,int y) //区间修改
{
    if(tree[k].l>=a&&tree[k].r<=b)
    {
        tree[k].w+=(tree[k].r-tree[k].l+1)*y;//(r-1)+1区间点的总数
        tree[k].f+=y;
        return;
    }
    if(tree[k].f)  down(k);
    int m=(tree[k].l+tree[k].r)>>1;
    if(a<=m)  change_interval(k<<1,a,b,y);
    if(b>m)  change_interval(k<<1|1,a,b,y);
    push_up(k);
}
int main()
{
    int a,b,x,y;
    char ss[5];
    while(scanf("%d%d",&n,&q)==2)
    {
        build(1,1,n);
        for(int i=1;i<=q;i++)
        {
            scanf("%s",ss);
            ans=0;
            if(strcmp(ss,"xx")==0)//单点查询
            {
                scanf("%d",&x);
                ask_point(1,x);
                printf("%d",ans);
            }
            else if(strcmp(ss,"xx")==0)//单点修改
            {
                scanf("%d%d",&x,&y);
                change_point(1,x,y);
            }
            else if(strcmp(ss,"xx")==0)//区间查询
            {
                scanf("%d%d",&a,&b);
                ans=0;
                ask_interval(1,a,b);
                printf("%d\n",ans);
            }
            else //区间修改
            {
                scanf("%d%d%d",&a,&b,&y);
                change_interval(1,a,b,y);
            }
        }
    }
    return 0;
}