线段树(二)——杨子曰算法

线段树(二)——杨子曰算法

传送门:线段树(一)


一个长度为n(n<=200000)的序列,还是m(m<=200000)个询问,对于每个询问先给出一个k,若k=1则输出x,y输出区间[x,y]的和,若k=2则输入x,y,将第x个数加上y


这个问题与之前最大的区别就是多了更新,这也就是线段树的优势了,杨子曰:线段树专注于在区间中更新的问题(怎么弄得跟企业的宣传标语一样)
也就是说,我们要在原来那个没有更新的代码的基础上增加一个过程update就行了

void update(int l,int r,int k,int v,int nod){
    //l,r表示目前线段树上的左右区间
    //k,v表示把第k个元素+=v
    //nod表示当前节点的编号 
}

从根节点开始往下走:
那么对于现在找到的节点就只有两种情况了
1.第k个元素在左区间,也就是要递归左子树,即update(l,mid,k,v,nod*2)
2.第k个元素在右区间,也就是要递归右子树,即update(mid+1,r,k,v,nod*2+1)
找到这个对应的叶子节点时,还要回溯,需要把整条路径上的值更改
那就完事了呗

void update(int l,int r,int k,int v,int nod){
    if(l==r){
        tree[nod]+=v;
        return;
    }
    int mid=(l+r)/2;
    if (k<=mid) update(l,mid,k,v,nod*2);
    else if (k>mid) update(mid+1,r,k,v,nod*2+1);
    tree[nod]=tree[nod*2]+tree[nod*2+1];
}

在上一道几乎一样的题:
题目描述
老管家是一个聪明能干的人。他为财主工作了整整10年,财主为了让自已账目更加清楚。要求管家每天记k次账,由于管家聪明能干,因而管家总是让财主十分满意。但是由于一些人的挑拨,财主还是对管家产生了怀疑。于是他决定用一种特别的方法来判断管家的忠诚,他把每次的账目按1,2,3…编号,然后不定时的问管家问题,问题是这样的:在a到b号账中最少的一笔是多少?为了让管家没时间作假他总是一次问多个问题。
在询问过程中账本的内容可能会被修改
输入
输入中第一行有两个数m,n表示有m(m<=100000)笔账,n表示有n个问题,n<=100000。
接下来每行为3个数字,第一个p为数字1或数字2,第二个数为x,第三个数为y
当p=1 则查询x,y区间
当p=2 则改变第x个数为y
输出
输出文件中为每个问题的答案。具体查看样例。
样例输入
10 3
1 2 3 4 5 6 7 8 9 10
1 2 7
2 2 0
1 1 10
样例输出
2 0

呵呵,完全一样嘛,就把刚才要算和的地方改成去min就欧了
上段代码:

#include<cstdio>
using namespace std;

int tree[400005],a[100005];

int min(int a,int b){
    return a<b?a:b;
}
void build(int l,int r,int nod){
    if (l==r){
        tree[nod]=a[l];
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,nod*2);
    build(mid+1,r,nod*2+1);
    tree[nod]=min(tree[nod*2],tree[nod*2+1]);
}

int query(int l,int r,int ll,int rr,int nod){
    if (l==ll && r==rr){
        return tree[nod];
    }
    int mid=(l+r)/2;
    if (rr<=mid) return query(l,mid,ll,rr,nod*2);
    else if (ll>=mid+1) return query(mid+1,r,ll,rr,nod*2+1);
    else return min(query(l,mid,ll,mid,nod*2),query(mid+1,r,mid+1,rr,nod*2+1));
}

void update(int l,int r,int k,int v,int nod){
    if (l==r){
        tree[nod]=v;
        return;
    }
    int mid=(l+r)/2;
    if (k<=mid) update(l,mid,k,v,nod*2);
    else update(mid+1,r,k,v,nod*2+1);
    tree[nod]=min(tree[nod*2],tree[nod*2+1]);
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    build(1,n,1);
    for (int i=1;i<=m;i++){
        int x,y,z;
        scanf("%d%d%d",&z,&x,&y);
        if (z==1){
            printf("%d\n",query(1,n,x,y,1));
        }
        else{
            update(1,n,x,y,1);
        }
    }
    return 0;
} 

OK 完事
现在更新的是一个数,如果变为更新一个区间What will happen?向下看↓:


线段树(三)预告:
给一段长度为n(n<=200000)的序列,接着是m(m<=20000)个操作对于每个操作显示一个k,如果k=1那么输入一个x,输出第x个元素,如果k=2,那么输入x,y,z,将区间[x,y]的每一个元素都加上z,88
传送门:线段树(三)

未经作者允许,严禁转载:https://blog.csdn.net/HenryYang2018/article/details/79880012

猜你喜欢

转载自blog.csdn.net/henryyang2018/article/details/79880012