线段树

淼润淽涵

于 2019-09-29 17:31:42 发布

阅读量134

点赞数

分类专栏：模板数据结构--线段树文章标签：数据结构--线段树模板

本文链接：https://blog.csdn.net/wcxyky/article/details/101703089

版权

模板同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

数据结构--线段树

4 篇文章 0 订阅

订阅专栏

线段树单点add，区间sum查询的模板

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50000+5;
 
//线段树需要维护的信息
int sum[maxn*4];
#define lson i*2, l, m
#define rson i*2+1, m+1, r
 
//i节点收集子节点的统计结果
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}
 
//递归建立线段树
void build(int i,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&sum[i]);
        return ;
    }
 
    int m=(l+r)/2;
    build(lson);
    build(rson);
    PushUp(i);//收集子节点的结果
}
 
//在当前区间[l, r]内查询区间[ql, qr]间的目标值
//且能执行这个函数的前提是：[l,r]与[ql,qr]的交集非空
//其实本函数返回的结果也是 它们交集的目标值
int query(int ql,int qr,int i,int l,int r)
{
    //目的区间包含当前区间
    if(ql<=l && r<=qr) return sum[i];
 
    int m=(l+r)/2;
    int res=0;
    if(ql<=m) res += query(ql,qr,lson);
    if(m<qr) res += query(ql,qr,rson);
    return res;
}
 
//update这个函数就有点定制的意味了
//本题是单点更新，所以是在区间[l,r]内使得第id数的值+val
//如果是区间更新，可以update的参数需要将id变为ql和qr
void update(int id,int val,int i,int l,int r)
{
    if(l==r)
    {
        sum[i] += val;
        return ;
    }
 
    int m=(l+r)/2;
    if(id<=m) update(id,val,lson);
    else update(id,val,rson);
    PushUp(i);//时刻记住维护i节点统计信息正确性
}

例题：HDU 1166 敌兵布阵(线段树:单点更新,区间求和)

http://acm.hdu.edu.cn/showproblem.php?pid=1166

题意：

给你n个整数，然后给你多条命令，每条命令如下格式：

(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 表示结束，这条命令在每组数据最后出现;

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

Case 1:
6
33
59

代码：

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50000+5;
 
//线段树需要维护的信息
int sum[maxn*4];
#define lson i*2, l, m
#define rson i*2+1, m+1, r
 
//i节点收集子节点的统计结果
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}
 
//递归建立线段树
void build(int i,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&sum[i]);
        return ;
    }
 
    int m=(l+r)/2;
    build(lson);
    build(rson);
    PushUp(i);//收集子节点的结果
}
 
//在当前区间[l, r]内查询区间[ql, qr]间的目标值
//且能执行这个函数的前提是：[l,r]与[ql,qr]的交集非空
//其实本函数返回的结果也是 它们交集的目标值
int query(int ql,int qr,int i,int l,int r)
{
    //目的区间包含当前区间
    if(ql<=l && r<=qr) return sum[i];
 
    int m=(l+r)/2;
    int res=0;
    if(ql<=m) res += query(ql,qr,lson);
    if(m<qr) res += query(ql,qr,rson);
    return res;
}
 
//update这个函数就有点定制的意味了
//本题是单点更新，所以是在区间[l,r]内使得第id数的值+val
//如果是区间更新，可以update的参数需要将id变为ql和qr
void update(int id,int val,int i,int l,int r)
{
    if(l==r)
    {
        sum[i] += val;
        return ;
    }
 
    int m=(l+r)/2;
    if(id<=m) update(id,val,lson);
    else update(id,val,rson);
    PushUp(i);//时刻记住维护i节点统计信息正确性
}
 
 
int main()
{
    int T;
    scanf("%d",&T);
    for(int kase=1; kase<=T; kase++)
    {
        printf("Case %d:\n",kase);
 
        int n;//节点总数
        scanf("%d",&n);
        build(1,1,n);//建立线段树
 
        char str[20];
        int u,v;
        while(scanf("%s",str)==1 && str[0]!='E')
        {
            scanf("%d%d",&u,&v);
 
            if(str[0]=='Q') printf("%d\n",query(u,v,1,1,n));
            else if(str[0]=='A') update(u,v,1,1,n);
            else update(u,-v,1,1,n);
        }
    }
    return 0;
}

例题：HDU-1754 I Hate It

Input

本题目包含多组测试，请处理到文件结束。
在每个测试的第一行，有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 )，分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数，代表这N个学生的初始成绩，其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取'Q'或'U') ，和两个正整数A，B。
当C为'Q'的时候，表示这是一条询问操作，它询问ID从A到B(包括A,B)的学生当中，成绩最高的是多少。
当C为'U'的时候，表示这是一条更新操作，要求把ID为A的学生的成绩更改为B。

Output

对于每一次询问操作，在一行里面输出最高成绩。

Sample Input
5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5
Sample Output
5
6
5
9
代码：
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX=2e5+10;
int n,m;
char str[3];
struct Node
{
	int l;
	int r;
	int Max;
}tree[MAX*4];
void Push(int o)
{
	  tree[o].Max=max(tree[o*2].Max,tree[o*2+1].Max);
}
void Buildtree(int o,int l,int r)
{
	    tree[o].l=l;
	    tree[o].r=r;
	    while(l==r)
	    {
            int t;
			scanf("%d",&t);
			tree[o].Max=t;
			return;
	    }
	   int mid=(l+r)/2;
	   Buildtree(o*2,l,mid);
       Buildtree(o*2+1,mid+1,r);
       Push(o);
}
int TreeMax(int o,int l,int r,int x,int y)
{
        if(l==x&&r==y)
        {
            return tree[o].Max;
        }
        int mid=(l+r)/2;
        if(y<=mid)
            return TreeMax(o*2,l,mid,x,y);
        else if(x>mid)
            return TreeMax(o*2+1,mid+1,r,x,y);
        else
            return max(TreeMax(o*2,l,mid,x,mid),TreeMax(o*2+1,mid+1,r,mid+1,y));
}
void Update(int o,int l,int r,int x,int y)
{
	   if(l==r)
	   {
	   	   tree[o].Max=y;
	   	   return;
	   }
	   int mid=(l+r)/2;
	   if(x<=mid)
	      Update(o*2,l,mid,x,y);
       else
	      Update(o*2+1,mid+1,r,x,y);
	    Push(o);
}
int main()
{
    while(~scanf("%d %d",&n,&m))
    {
          Buildtree(1,1,n);
          while(m--)
    	  {
    	 	 int a,b;
    	 	 scanf("%s %d %d",str,&a,&b);
    	 	 if(str[0]=='Q')
             {
    	 	  	    printf("%d\n",TreeMax(1,1,n,a,b));
             }
             else
			    Update(1,1,n,a,b);
		 }
    }
	return 0;
}

延迟更新版的线段树区间add和区间求和的模板

//区间add，区间查询
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
 
//每当有add加到i节点上，不会去更新i节点的sum.
//也就是说如果要查询区间[1,n]的sum值，既要考虑sum[i]的值，也要考虑add[i]的值
const int MAXN=100000+100;
typedef long long LL;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
int sum[MAXN*4];
int addv[MAXN*4];
 
//i节点收集下面节点的信息
void PushUp(int i,int num)
{
    sum[i]=sum[i*2]+sum[i*2+1]+addv[i*2]*(num+1)/2+addv[i*2+1]*(num-(num+1)/2);
}
 
//将i节点的addv压下去，且更新sum[i]
void PushDown(int i,int num)
{
    if(addv[i])
    {
        sum[i] += addv[i]*num;
        addv[i*2]+=addv[i]; addv[i*2+1]+=addv[i];
        addv[i]=0;
    }
}
 
//sum[i]收集子节点的信息
//只在build线段树时会用
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}
void build(int i,int l,int r)
{
    addv[i]=0;
    if(l==r)
    {
        scanf("%I64d",&sum[i]);
        return ;
    }
    int m=(l+r)/2;
    build(lson);
    build(rson);
    PushUp(i,r-l+1);
}
 
//给[ql,qr]区间加上add值
void update(int ql,int qr,int add,int i,int l,int r)
{
    if(ql<=l&&r<=qr)
    {
        addv[i]+=add;
        //sum[i] += (LL)add*(r-l+1);
        return ;
    }
    PushDown(i,r-l+1);
    int m=(l+r)/2;
    if(ql<=m) update(ql,qr,add,lson);
    if(m<qr) update(ql,qr,add,rson);
    PushUp(i,r-l+1);
}
 
//查询[ql,qr]区间的sum值
int query(int ql,int qr,int i,int l,int r)
{
    if(ql<=l&&r<=qr)
    {
        PushDown(i,r-l+1);
        return sum[i];
    }
    PushDown(i,r-l+1);
    int m=(l+r)/2;
    int res=0;
    if(ql<=m) res+=query(ql,qr,lson);
    if(m<qr) res+=query(ql,qr,rson);
    return res;
}

例题：POJ 3468 A Simple Problemwith Integers(线段树:区间add,区间查询)

http://poj.org/problem?id=3468

题意：

You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

Input

The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A1, A2, ... , AN. -1000000000 ≤ Ai ≤ 1000000000.
Each of the next Q lines represents an operation.
"C a b c" means adding c to each of Aa, Aa+1, ... , Ab. -10000 ≤ c ≤ 10000.
"Q a b" means querying the sum of Aa, Aa+1, ... , Ab.

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

代码：

//POJ 3468 区间add，区间查询
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
 
//每当有add加到i节点上，直接更新i节点的sum.
//也就是说如果要查询区间[1,n]的sum值，直接sum[1]即可,不用再去考虑1的addv[1]值.
const int MAXN=100000+100;
typedef long long LL;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
LL sum[MAXN*4];
LL addv[MAXN*4];
void PushDown(int i,int num)
{
    if(addv[i])
    {
        sum[i*2] +=addv[i]*(num-(num/2));
        sum[i*2+1] +=addv[i]*(num/2);
        addv[i*2] +=addv[i];
        addv[i*2+1]+=addv[i];
        addv[i]=0;
    }
}
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}
void build(int i,int l,int r)
{
    addv[i]=0;
    if(l==r)
    {
        scanf("%I64d",&sum[i]);
        return ;
    }
    int m=(l+r)/2;
    build(lson);
    build(rson);
    PushUp(i);
}
void update(int ql,int qr,int add,int i,int l,int r)
{
    if(ql<=l&&r<=qr)
    {
        addv[i]+=add;
        sum[i] += (LL)add*(r-l+1);
        return ;
    }
    PushDown(i,r-l+1);
    int m=(l+r)/2;
    if(ql<=m) update(ql,qr,add,lson);
    if(m<qr) update(ql,qr,add,rson);
    PushUp(i);
}
LL query(int ql,int qr,int i,int l,int r)
{
    if(ql<=l&&r<=qr)
    {
        return sum[i];
    }
    PushDown(i,r-l+1);
    int m=(l+r)/2;
    LL res=0;
    if(ql<=m) res+=query(ql,qr,lson);
    if(m<qr) res+=query(ql,qr,rson);
    return res;
}
int main()
{
    int n,q;
    while(scanf("%d%d",&n,&q)==2&&n&&q)
    {
        build(1,1,n);
        while(q--)
        {
            char str[10];
            scanf("%s",str);
            int x,y,z;
            if(str[0]=='Q')
            {
                scanf("%d%d",&x,&y);
                printf("%I64d\n",query(x,y,1,1,n));
            }
            else
            {
                scanf("%d%d%d",&x,&y,&z);
                update(x,y,z,1,1,n);
            }
        }
    }
    return 0;
}

淼润淽涵

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
线段树

线段树单点add，区间sum查询的模板#include<cstdio>#include<cstring>using namespace std;const int maxn=50000+5; //线段树需要维护的信息int sum[maxn*4];#define lson i*2, l, m#define rson i*2+1, m+1, r ...
复制链接

扫一扫

专栏目录