线段树初步(1)

蒟蒻终于要开始好好学线段树了……

线段树是一种二叉树形结构(二叉搜索树),属于平衡树的一种。它将线段区间组织成树形的结构,并用每个节点来表示一条线段[a,b]。每个节点的左右儿子线段分别是该线段的左半[a,(a+b)/2]和右半[(a+b)/2+1,b]区间,递归定义之后就是一棵线段树。

1.建树

既然线段树是递归建立的,那么我们就在递归到叶节点的时候把数据植入,之后递归返回的时候,将要修改的节点的值修改为其子节点的值之和即可。

void build(int l,int r,int x)//递归建树,到叶节点输入,之后递归返回修改
{
    if(l == r)
    {
        tree[x] = read();
        return;
    }
    int mid = (l+r) >> 1;
    build(l,mid,x << 1);
    build(mid+1,r,x << 1 | 1);
    tree[x] = tree[x << 1] + tree[x << 1 | 1];
} 

2.单点修改(增加或者删除)

因为每条节点表示一条线段……所以在单点修改的时候我们只要找到需要修改的节点所在的位置,之后当当前区间l==r的时候修改这个点就可以。如果要删除,就往里面传负数就行。

注意修改结束之后返回的时候也要……递归把其父亲和祖先的值都修改一遍。

void update(int k,int val,int l,int r,int x)//单点修改
{
    if(l == r)
    {
        tree[x] += val;
        return;
    }
    int mid = (l+r) >> 1;
    if(k <= mid) update(k,val,l,mid,x << 1);
    else update(k,val,mid+1,r,x << 1 | 1);
    tree[x] = tree[x << 1] + tree[x << 1 | 1];
}//k为要修改的节点编号,val为修改值,l,r为左右区间

3.区间查询

给定一段区间,求区间中所有元素之和。

当我们当前访问的区间其被完全包含于所求区间的时候,直接把这段区间和加上即可。

否则的话我们进行二分,把所求区间分割成更小的区间,之后找到能被其完全包含的区间进行累加即可。(也就是相当于我们把所求区间分割成了连续的小区间之后求和)

累加的过程同样是在递归中完成的。

int query(int kl,int kr,int l,int r,int x)
{
    if(kl <= l && kr >= r) return tree[x];
    int mid = (l+r) >> 1;
    int sum = 0;
    if(kl <= mid) sum += query(kl,kr,l,mid,x << 1);
    if(kr > mid) sum += query(kl,kr,mid+1,r,x << 1 | 1);
    return sum;
} //kl为查询左端点,kr为查询右端点,l,r为当前区间作用端点,x为节点编号 

会了这几个操作之后就可以做一道稍微简单一点的练习题了。

C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人。

【输入格式】

第一行一个整数T,表示有T组数据。
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有4种形式:
(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 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令

【输出格式】

对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

这道题为了方便我把字符串都改成了数字。

操作就三个,建树,单点修改,区间查询,上面三个操作正好够用。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<queue>
#define rep(i,a,n) for(int i = a;i <= n;i++)
#define per(i,n,a) for(int i = n;i >= a;i--)
#define enter putchar(‘\n‘)

using namespace std;
const int M = 1000005;
typedef long long ll;

int tree[M*4],k = 1,c,t,n,p,q,num;
int read()
{
    int ans = 0,op = 1;
    char ch = getchar();
    while(ch < ‘0‘ || ch > ‘9‘)
    {
        if(ch == ‘-‘) op = -1;
        ch = getchar();
    }
    while(ch >= ‘0‘ && ch <= ‘9‘)
    {
        ans *= 10;
        ans += ch - ‘0‘;
        ch = getchar();
    }
    return ans * op;
}
void build(int l,int r,int x)//递归建树,到叶节点输入,之后递归返回修改
{
    if(l == r)
    {
        tree[x] = read();
        return;
    }
    int mid = (l+r) >> 1;
    build(l,mid,x << 1);
    build(mid+1,r,x << 1 | 1);
    tree[x] = tree[x << 1] + tree[x << 1 | 1];
}
void update(int k,int val,int l,int r,int x)//单点修改
{
    if(l == r)
    {
        tree[x] += val;
        return;
    }
    int mid = (l+r) >> 1;
    if(k <= mid) update(k,val,l,mid,x << 1);
    else update(k,val,mid+1,r,x << 1 | 1);
    tree[x] = tree[x << 1] + tree[x << 1 | 1];
}//k为要修改的节点编号,val为修改值,l,r为左右区间
int query(int kl,int kr,int l,int r,int x)
{
    if(kl <= l && kr >= r) return tree[x];
    int mid = (l+r) >> 1;
    int sum = 0;
    if(kl <= mid) sum += query(kl,kr,l,mid,x << 1);
    if(kr > mid) sum += query(kl,kr,mid+1,r,x << 1 | 1);
    return sum;
} //kl为查询左端点,kr为查询右端点,l,r为当前区间作用端点,x为节点编号
int main()
{
    t = read();
    while(t--)
    {
        n = read();
        build(1,n,1);
        c = read();
        rep(i,1,c)
        {
            num = read();
            if(num == 4) break;
            p = read(),q = read();
            if(num == 1) update(p,q,1,n,1);
            if(num == 2) update(p,-q,1,n,1);
            if(num == 3) printf("%d\n",query(p,q,1,n,1));
        }
    }
    return 0;
}
/*
1
10
1 2 3 4 5 6 7 8 9 10
7
3 1 3
1 3 6
3 2 7
2 10 2
1 6 3
3 3 10
4
*/

原文地址:https://www.cnblogs.com/captain1/p/8996763.html

时间: 2024-11-08 02:40:11

线段树初步(1)的相关文章

线段树初步&amp;&amp;lazy标记

线段树 一.概述: 线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点. 对于线段树中的每一个非叶子节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b].因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度. 使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN).而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩. 二.基本操作: 1

HDU 1754 I Hate It(线段树初步应用)

HDU  1754  I Hate It Description 很多学校流行一种比较的习惯.老师们很喜欢询问,从某某到某某当中,分数最高的是多少. 这让很多学生很反感. 不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问.当然,老师有时候需要更新某位同学的成绩. Input 本题目包含多组测试,请处理到文件结束. 在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目. 学生ID

线段树初步

线段树模板1:https://www.luogu.org/problem/show?pid=3372 线段树模板2:https://www.luogu.org/problem/show?pid=3373 这些都比较基础,就是1或2个lazy标记的时候怎么处理?几乎不用考虑兼容性的问题. 现在这里有一道充分考验线段树lazy的兼容性问题的题目,涉及到4个lazy标记,怎么处理? 例子1:求线段树维护一个区间,支持如下操作:区间修改为同一个值(更改1),区间加一个数(更改2),区间和(运算1),区间

可持久化线段树(待补充)

可持久化线段树初步理解(单点修改): 当需要同时保留修改前和修改后数据时,可能就要用到可持久化数据结构.考虑线段树的单点修改,实际上只改了一部分节点的值(logn级别),如果重建一棵树,需要nlogn级别的时间和空间.此时,我们可以使用可持久化线段树,将需要被修改的节点建成新的点,并且每次修改给予线段树一个新的根,这样每次修改并且保存历史记录只用logn级别的时间,多开logn级别的空间. 我的实现:线段树使用结构体Node{ int lson,int rson; int val; }; 代码与

关于线段树的初步理解

POJ 3264 根据题目意思就是给定一段序列,然后给出几个区间,要求出该段区间中最大值与最小值之差. 首先我想到的是用数组存储这一段序列,然后每次根据区间的左右边界来遍历这个段序列然后找到最大值和最小值,显然这样的方法是最容易想到的,但是可想而知这样的方法会耗费很多的时间,时间复杂度太大. 然后,我进行了第二种思考,就是利用二维数组来存储左边界到右边界的最大值: (1)如果l==r,那么显然当前区间的最大值和最小值就是l这个位置的数本身 (2)max1[i][j]=max{max[i][j-1

HDU 1166 敌兵布阵(线段树的初步应用2)

HDU  1166  敌兵布阵 Description C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了.A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况.由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视. 中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工

【BZOJ-3779】重组病毒 LinkCutTree + 线段树 + DFS序

3779: 重组病毒 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 224  Solved: 95[Submit][Status][Discuss] Description 黑客们通过对已有的病毒反编译,将许多不同的病毒重组,并重新编译出了新型的重组病毒.这种病毒的繁殖和变异能力极强.为了阻止这种病毒传播,某安全机构策划了一次实验,来研究这种病毒.实验在一个封闭的局域网内进行.局域网内有n台计算机,编号为1~n.一些计算机之间通过网线直接相连,形

Luck and Love(二维线段树)

Luck and Love Time Limit: 10000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 54 Accepted Submission(s): 21   Problem Description 世界上上最远的距离不是相隔天涯海角而是我在你面前可你却不知道我爱你                ―― 张小娴 前段日子,枫冰叶子给Wiskey做了个征婚启事,聘

SDUSTOJ 1796 哆啦A梦的军队(线段树维护前缀位置)

Description 在2050年机器人战争爆发,聪明的机器猫为了帮助大雄打赢这场战 争,从自己口袋里掏出了机器人战棋,每一个战棋都可以成为一名战士,哆啦A梦决定给他们整整队,哆啦A梦发现第 i 个位置的战士编号为 Ai(显然 A 是一个排列).经过计算,哆啦A梦发现,让第 i 个位置的战士编号为 Bi 时,他的军队可以发挥出最大的战斗力(保证 B 也是一个排列). 哆啦A梦可以发出指令来改变战士们的排列顺序,每一次,他都会报出一个整数 i(1≤i<n).如果排在第 i 个位置的战士编号大于第