线段树 weed

来源:互联网 发布:网络机房装修标准 编辑:程序博客网 时间:2024/06/05 18:02

有两个操作:1,加上一层厚度为vi的土;2,挖掉vi层土。
先给出m个操作,之后再给出q个询问,问把第ci个操作改成什么操作后,最终剩下土的厚度。
m<=2*10^5, q<=2*10^5
最近做了好多线段树啊。。。(但考试时压根想不到这是线段树)
尤其是这道题。
记录这个节点所在区间要向左边删几个,留下了几个,留下值的总和。
那么问题就在于对左右子节点的合并了。因为记录的左右子节点的删除个数,剩余个数,那么只要判断是左边剩下的多,还是右边删得多。如果删得多,那处理就比较方便了。(自己想。。。);但如果是剩的多呢?那么左儿子会剩下一部分,这个就要递归处理了,删掉右儿子的删除的个数。那么求这个值我们没必要每一次都递归,可以多维护一个变量,存下来,再不修改的情况下,减少计算量(其实如果不维护会TLE)

#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#include <algorithm>#include <cmath>#define N 200005#define ll long longusing namespace std;int read(){    int sum=0,f=1;char x=getchar();    while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}    while(x>='0'&&x<='9'){sum=(sum<<1)+(sum<<3)+x-'0';x=getchar();}    return sum*f;}struct node{int k,v;}a[N];int m,q;namespace TREE{    struct tree    {        int l,r,sum,del,add,s;    }t[N*4];    int dell(int x,int k)    {        if(t[x*2+1].add==k)return t[x*2].s;        if(t[x*2+1].add>k)return t[x*2].s+dell(x*2+1,k);        return dell(x*2,k-t[x*2+1].add+t[x*2+1].del);    }    void up(int x)    {        t[x].sum=0;        if(t[x*2].add>t[x*2+1].del)        {            int k=t[x*2].add-t[x*2+1].del;            t[x].add=t[x*2+1].add+k;            t[x].del=t[x*2].del;            if(t[x*2+1].del!=0)t[x*2].s=dell(x*2,t[x*2+1].del);            else t[x*2].s=t[x*2].sum;            t[x].sum=t[x*2+1].sum+t[x*2].s;        }           else        {            t[x].add=t[x*2+1].add;            t[x].del=t[x*2].del+t[x*2+1].del-t[x*2].add;            t[x*2].s=0;            t[x].sum=t[x*2+1].sum;        }    }    inline void build(int l,int r,int x)    {        t[x].l=l;t[x].r=r;        if(l==r)        {            if(a[l].k==0){t[x].add=1;t[x].sum=t[x].s=a[l].v;}            else{t[x].del=a[l].v;}            return;        }        int mid=l+r>>1;        build(l,mid,x*2);build(mid+1,r,x*2+1);        up(x);    }    inline void C(int l,int x)    {        if(t[x].l==t[x].r)        {            if(a[l].k==0){t[x].del=0;t[x].add=1;t[x].sum=t[x].s=a[l].v;}            else{t[x].del=a[l].v;t[x].add=0;t[x].sum=t[x].s=0;}            return;        }        int mid=t[x].l+t[x].r>>1;        if(l<=mid)C(l,x*2);else C(l,x*2+1);        up(x);    }}using namespace TREE;int main(){    m=read();q=read();    for(int i=1;i<=m;i++)        a[i].k=read(),a[i].v=read();    build(1,m,1);int c;    while(q--)    {        c=read();a[c].k=read();a[c].v=read();        C(c,1);        printf("%d\n",t[1].sum);    }}
原创粉丝点击