weed(线段树)

题面

 
 
 
 
 
 
 
 

解:

这道题学习到了线段树的新姿势,啊啊啊啊,还是自己太菜了。
这道题告诉我们线段树除了O1合并之外,在数据可以 log2 的情况下,我们可以考虑 log 合并。
这道题如何维护。很容易想到,我们维护区间不被删除的和,区间还剩多少堆。但是我们发现这样不好update,不知道删除操作具体对那些堆产生影响。于是我们维护一个区间溢出删除个数,就是说这个区间里的删除操作还会向前删多少个。然后我们update的时候就用右儿子溢出值查询一下左儿子从右向左的和。总复杂度是 log2 的。
看一下update:

int query(int u,int x)
{
    if(b[b[u].rson].kkk==x)
      return b[u].num-b[b[u].rson].num;
    if(x<b[b[u].rson].kkk)
      return b[u].num-b[b[u].rson].num+query(b[u].rson,x);
    else
      return query(b[u].lson,x-b[b[u].rson].kkk+b[b[u].rson].out);
//  if(b[b[u].rson].kkk==x) return b[b[u].rson].num;
//  if(b[b[u].rson].kkk>x)
//    return query(b[u].rson,x);
//  else return b[b[u].rson].num+query(b[u].lson,x+b[b[u].rson].out-b[b[u].rson].kkk);
}

void update(int u)
{
    int l=b[u].lson,r=b[u].rson;
    if(b[r].out==0)
    {
        b[u].num=b[l].num+b[r].num;
        b[u].kkk=b[l].kkk+b[r].kkk;
        b[u].out=b[l].out;
        return;
    }
    if(b[r].out>=b[l].kkk)
    {
        b[u].num=b[r].num;
        b[u].kkk=b[r].kkk;
        b[u].out=b[r].out-b[l].kkk+b[l].out;
        return;
    }
    b[u].num=b[r].num+query(l,b[r].out);
    b[u].out=b[l].out;
    b[u].kkk=b[r].kkk+b[l].kkk-b[r].out;
    return;
}

一开始使用的被注释的代码,但是这样是WA的,因为如果查询从右边数x个的和,不能用总数减,有些节点会重复算,因为在上一层的节点就已经把这些点抛掉了,这样相当于又减了一次。不加注释才是正确的。

最后还是贴一下代码吧:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct lxy{
    int lson,rson,l,r,out,num,kkk;
}b[400005];
int n,q,root,cnt,id,k,v;
struct lxy1{
    int k,v;
}data[200005];

void upup()
{
    if(k==1)
    {
        b[id].num=0;
        b[id].out=v;
        b[id].kkk=0;
    }
    if(k==0)
    {
        b[id].num=v;
        b[id].out=0;
        b[id].kkk=1;
    }
    return;
}

int query(int u,int x)
{
    if(b[b[u].rson].kkk==x)
      return b[u].num-b[b[u].rson].num;
    if(x<b[b[u].rson].kkk)
      return b[u].num-b[b[u].rson].num+query(b[u].rson,x);
    else
      return query(b[u].lson,x-b[b[u].rson].kkk+b[b[u].rson].out);
}

void update(int u)
{
    int l=b[u].lson,r=b[u].rson;
    if(b[r].out==0)
    {
        b[u].num=b[l].num+b[r].num;
        b[u].kkk=b[l].kkk+b[r].kkk;
        b[u].out=b[l].out;
        return;
    }
    if(b[r].out>=b[l].kkk)
    {
        b[u].num=b[r].num;
        b[u].kkk=b[r].kkk;
        b[u].out=b[r].out-b[l].kkk+b[l].out;
        return;
    }
    b[u].num=b[r].num+query(l,b[r].out);
    b[u].out=b[l].out;
    b[u].kkk=b[r].kkk+b[l].kkk-b[r].out;
    return;
}

void build(int &u,int l,int r)
{
    u=++cnt;
    b[u].l=l;b[u].r=r;
    if(l==r)
    {
        id=u,k=data[l].k,v=data[l].v;
        upup();
        return;
    }
    int mid=(l+r)>>1;
    build(b[u].lson,l,mid);
    build(b[u].rson,mid+1,r);
    update(u);
}

void modify(int u)
{
    if(b[u].l==b[u].r)
    {
        id=u;
        upup();
        return;
    }
    int mid=(b[u].l+b[u].r)>>1;
    if(id<=mid) modify(b[u].lson);
    else modify(b[u].rson);
    update(u);
}

int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
      scanf("%d%d",&data[i].k,&data[i].v);
    build(root,1,n);
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d%d",&id,&k,&v);
        modify(root);
        printf("%d\n",b[root].num);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值