洛谷篇 线段树模板

区间求和

需要用线段树这样的数据结构来储存数据,从而降低操作的时间复杂度。

模板说明:

输出:1.区间求和

 2.区间求极值

操作:单点修改、区间修改

const int inf=0x3f3f3f3f;

const int maxn=1e5+5;

Long long sum[maxn*4],minn[maxn*4],maxx[maxn*4],vis[maxn*4]

Int Map[maxn],n;

//sum储存和,minn最小值,maxx最大值,vis作为求最值的标记数组,Map是原数组

//建树

void built(int l,int r,int root)

{

if(l==r)         //叶子结点

{

vis[root]=0;

        sum[root]=Map[l];

        maxx[root]=Map[l];

        minn[root]=Map[l];

        return;

}

int mid=(l+r)/2;

    built(l,mid,root*2);          //创建左子树

    built(mid+1,r,root*2+1);      //创建右子树

    sum[root]=sum[root*2]+sum[root*2+1];         //根节点的和等于左右子树的和

    maxx[root]=max(maxx[root*2],maxx[root*2+1]);

    minn[root]=min(minn[root*2],minn[root*2+1]);

    vis[root]=0;

}

//更新

void push_up(int l,int r,int root,int x)  

{

vis[root]+=x;//+号

    sum[root]+=x*(r-l+1);//同时加数

    maxx[root]=x;

    minn[root]=x;

}

void push_down(int l,int r,int root)

{

if(vis[root]==0)

return;

    int mid=(l+r)/2;

    push_up(l,mid,root*2,vis[root]);

    push_up(mid+1,r,root*2+1,vis[root]);

    vis[root]=0;

}

//区间求和

long long query_sum(int L,int R,int l,int r,int root)

{

if(L<=l&&r<=R)          //区间包含

    return sum[root];

    push_down(l,r,root);

    int mid=(l+r)/2;

    long long sum1=0,sum2=0;

    if(L<=mid)

    sum1=query_sum(L,R,l,mid,root*2);             //区间在左子树

    if(R>mid)

    sum2=query_sum(L,R,mid+1,r,root*2+1);        //在右子树

    return sum1+sum2;

}

//区间修改

void update(int L,int R,int x,int l,int r,int root)

{

if(L<=l&&R>=r)

{

push_up(l,r,root,x);

return ;

}

push_down(l,r,root);              //向上更新

    int mid=(l+r)/2;

    if(L<=mid)

    update(L,R,x,l,mid,root*2);

    if(R>mid)

    update(L,R,x,mid+1,r,root*2+1);

    sum[root]=sum[root*2]+sum[root*2+1];

    maxx[root]=max(maxx[root*2],maxx[root*2+1]);

    minn[root]=min(minn[root*2],minn[root*2+1]);

}

//单点修改

void Update(int k,int x,int l,int r,int root)

{

if(l==r)

{

        Map[l]=x;

        sum[root]=x;

        maxx[root]=x;

        minn[root]=x;

        return;

    }

    int mid=(l+r)/2;

    if(k<=mid)

    Update(k,x,l,mid,root*2);

    if(k>mid)

    Update(k,x,mid+1,r,root*2+1);

    sum[root]=sum[root*2]+sum[root*2+1];

    maxx[root]=max(maxx[root*2],maxx[root*2+1]);

    minn[root]=min(minn[root*2],minn[root*2+1]);

}

//求极值

int query_max(int L,int R,int l,int r,int root)

{

if(L<=l&&r<=R)

    return maxx[root];

    push_down(l,r,root);

    int mid=(l+r)/2;

    int t1=-inf,t2=-inf;

    if(L<=mid)

t1=query_max(L,R,l,mid,root*2);

    if(R>mid)

    t2=query_max(L,R,mid+1,r,root*2+1);

    return max(t1,t2);

}

int query_min(int L,int R,int l,int r,int root)

{

if(L<=l&&r<=R)

    return minn[root];

    push_down(l,r,root);

    int mid=(l+r)/2;

    int t1=inf,t2=inf;

    if(L<=mid)

t1=query_min(L,R,l,mid,root*2);

    if(R>mid)

    t2=query_min(L,R,mid+1,r,root*2+1);

    return min(t1,t2);

}

P3372

# 【模板】线段树 1

## 题目描述

如题,已知一个数列,你需要进行下面两种操作:

1. 将某区间每一个数加上 $k$。
2. 求出某区间每一个数的和。

## 输入格式

第一行包含两个整数 $n, m$,分别表示该数列数字的个数和操作的总个数。

第二行包含 $n$ 个用空格分隔的整数,其中第 $i$ 个数字表示数列第 $i$ 项的初始值。

接下来 $m$ 行每行包含 $3$ 或 $4$ 个整数,表示一个操作,具体如下:

1. `1 x y k`:将区间 $[x, y]$ 内每个数加上 $k$。
2. `2 x y`:输出区间 $[x, y]$ 内每个数的和。

## 输出格式

输出包含若干行整数,即为所有操作 2 的结果。

## 样例 #1

### 样例输入 #1

```
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
```

### 样例输出 #1

```
11
8
20
```

## 提示

对于 $30\%$ 的数据:$n \le 8$,$m \le 10$。  
对于 $70\%$ 的数据:$n \le {10}^3$,$m \le {10}^4$。  
对于 $100\%$ 的数据:$1 \le n, m \le {10}^5$。

保证任意时刻数列中任意元素的和在 $[-2^{63}, 2^{63})$ 内。

**【样例解释】**

![](https://cdn.luogu.com.cn/upload/pic/2251.png)

参考: 线段树模板_Ever_glow的博客-CSDN博客 

#include<iostream>
#include<stdio.h>
#include <algorithm>
#include<string>
#include <bits/stdc++.h>
using namespace std;

const int maxn=100010;
int sum[maxn*4],vis[maxn*4];
int Map[maxn],n;
void built(int l,int r,int root)
{
	if(l==r)         //叶子结点 
	{

        sum[root]=Map[l];

        return;
	}
	int mid=(l+r)/2;
    built(l,mid,root*2);          //创建左子树 
    built(mid+1,r,root*2+1);      //创建右子树 
    sum[root]=sum[root*2]+sum[root*2+1];         //根节点的和等于左右子树的和 

}
//更新
void push_up(int l,int r,int root,int x)  
{
	vis[root]+=x;
    sum[root]+=x*(r-l+1);

}
 
void push_down(int l,int r,int root)
{
	if(vis[root]==0)
	return;
    int mid=(l+r)/2;
    push_up(l,mid,root*2,vis[root]);
    push_up(mid+1,r,root*2+1,vis[root]);
    vis[root]=0;
}
//区间求和
int query_sum(int L,int R,int l,int r,int root)
{
	if(L<=l&&r<=R)          //区间包含 
    return sum[root];
    push_down(l,r,root);
    int mid=(l+r)/2;
    int sum1=0,sum2=0;
    if(L<=mid)
    sum1=query_sum(L,R,l,mid,root*2);             //区间在左子树
    if(R>mid)
    sum2=query_sum(L,R,mid+1,r,root*2+1);        //在右子树
    return sum1+sum2;
}
//区间修改
void update(int L,int R,int x,int l,int r,int root)
{
	if(L<=l&&R>=r)
	{
//		sum[root]+=(R-L+1)*x;
		push_up(l,r,root,x);
		return ;
	}
	push_down(l,r,root);              //向上更新 
    int mid=(l+r)/2;
    if(L<=mid)
    update(L,R,x,l,mid,root*2);
    if(R>mid)
    update(L,R,x,mid+1,r,root*2+1);

    sum[root]=sum[root*2]+sum[root*2+1];

}

int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1	;i<=n;i++)
	{
		cin>>Map[i];
	 } 
	 built(1,n,1);
	for(int i=1;i<=m;i++)
	{
		int k,a,b,t,y,z; 
		cin>>k;
		if(k==1)
		{
			cin>>a>>b>>t;
			update(a,b,t,1,n,1);
		}
		if(k==2)
		{
			cin>>y>>z;
			cout<<query_sum(y,z,1,n,1)<<endl;
		}
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值