线段树及主席树---板子题

NP_hard

已于 2022-01-29 19:18:35 修改

阅读量319

点赞数 1

分类专栏： Data Structure 文章标签：几何学

于 2020-12-19 22:51:32 首次发布

本文链接：https://blog.csdn.net/NP_hard/article/details/111411545

版权

文章目录

线段树及主席树

个人理解

线段树理解

由于只是一个个人笔记，故这里不再赘述
参考博客

线段树模板

#include<iostream>
#include<string.h>
#define max 1000
#define INFINITY 9999999
using namespace std;

struct node
{
   
    int data;
    int l,r;
    int sum;
    int min;
    int lazy;//延迟标记
    int addmark;//
}SegTree[max];

int arr[max]={
   0};

int min(int a,int b){
   
    return a<b?a:b;
}

void pushdown(int root)
{
   
    if(SegTree[root].addmark!=0){
   
        //设置左右孩子的标志域,孩子节点可能被多次延迟
        SegTree[root*2].addmark+=SegTree[root].addmark;
        SegTree[root*2+1].addmark+=SegTree[root].addmark;
        //更新
        SegTree[root*2].data+=SegTree[root].addmark;
        SegTree[root*2+1].data+=SegTree[root].addmark;
        SegTree[root].addmark=0;//清空
    }
}

void bulid(int root,int arr[],int l,int r)//区间最小值
{
   
    SegTree[root].l=l,SegTree[root].r=r;
    if(l==r){
   
        SegTree[root].data=arr[l];
        SegTree[root].sum=arr[l];
    }
    else{
      
        int mid=(l+r)/2;
        bulid(root*2,arr,l,mid);//递归构造左子树
        bulid(root*2+1,arr,mid+1,r);
        SegTree[root].data=min(SegTree[root*2].data,SegTree[root*2+1].data);
        SegTree[root].sum=SegTree[root*2].sum+SegTree[root*2+1].sum;
    }
}

//区间查询
int query(int k,int l,int r)//l~r<=k.l~k.r
{
   
    if(SegTree[k].lazy)//下传懒惰标记
        pushdown(k);
    if(SegTree[k].l==l&&SegTree[k].r==r)//区间正好
        return SegTree[k].sum;
    int mid=(SegTree[k].l+SegTree[k].r)/2;
    if(r<=mid)
        return query(k*2,l,r);
    if(l>mid)
        return query(k*2+1,l,r);
    return query(k*2,l,mid)+query(k*2+1,mid+1,r);
}

//区间更新
void update(int i,int l,int r,int k)
{
   
    if(SegTree[i].r<=r && SegTree[i].l>=l)//如果当前区间被完全覆盖在目标区间里，讲这个区间的sum+k*(tree[i].r-tree[i].l+1)
    {
   
        SegTree[i].sum+=k*(SegTree[i].r-SegTree[i].l+1);
        SegTree[i].lazy+=k;//记录lazytage
        return ;
    }
    pushdown(i);//向下传递
    if(SegTree[i*2].r>=l)
        update(i*2,l,r,k);
    if(SegTree[i*2+1].l<=r)
        update(i*2+1,l,r,k);
    SegTree[i].sum=SegTree[i*2].sum+SegTree[i*2+1].sum;
    return ;
}

//单点更新
void add(int i,int dis,int k)
{
   
    if(SegTree[i].l==SegTree[i].r){
   
        SegTree[i].sum+=k;
        return;
    }
    if(dis<=SegTree[i*2].r) add(i*2,dis,k);
    else add(i*2+1,dis,k);
    SegTree[i].sum=SegTree[i*2].sum+SegTree[i*2+1].sum;
    return;
}

int main()
{
   
    memset(SegTree,0,sizeof(SegTree));
    arr[1]=4,arr[2]=3,arr[3]=2,arr[4]=1,arr[5]=5,arr[6]=6;
    bulid(1,arr,1,6);
    cout << query(1,1,6) << endl;
    add(1,1,2);
    cout << query(1,1,6) << endl;
    update(1,1,6,1)