【模拟8.10】Weed(线段树)

考试只好随便骗骗分过去啦啦啦.....

正解是玄学线段树：

以每个操作为叶子节点，我们定义几个变量ce表示层数，h表示高度，add表示所减的层数

那么问题转化为单点修改的问题输出直接是根节点答案

但是我们发现合并区间很毒瘤

我们分两种情况：

设L为左儿子，R为右儿子。

1.T[L].ce<T[R].add 显然我们只需O(1)修改。

2.T[L].ce>T[R].add 我们需要将左子树后面的add个减去，我们自然想到了类似平衡树的

查找后缀的操作,于是肝.....

开始时我的query计算有多少减去，但是WA20

因为少考虑一种情况：

你的左子树中假设左子树根结点为K,左儿子Lson，右儿子Rson

右儿子中也可能有剩余的add，

那么我们直接发现右面不够查左面是不行的.........

听了toot大神的建议，我改为了查询还有多少留下

     if(t[k*2+1].ce<ceng)
     {
          qq_sum+=t[k*2+1].h;
          query_sum(k*2,l,mid,ceng-t[k*2+1].ce+t[k*2+1].add);
     }

当发现右端点少时，向左端点转移还需要减去靠右的add个

然后就没啥了，注意细节就好

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<map>
#define int long long
#define MAXN 1000010
using namespace std;
int read()
{
    int x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')
    {
         x=(x<<1)+(x<<3)+(c^48);
         c=getchar();
    }
    return x;
}
struct node{int ce,h,add,l,r;}t[MAXN*2];
int qq_sum;int m,q;
int orz[MAXN],mm[MAXN];
void query_sum(int k,int l,int r,int ceng)
{
     if(l==r)
     {
        if(ceng!=0)qq_sum+=t[k].h;
        return ;
     }
     int mid=(l+r)>>1;
     //printf("k=%lld l=%lld r=%lld
",k,l,r);
     if(t[k*2+1].ce<ceng)
     {
          qq_sum+=t[k*2+1].h;
          query_sum(k*2,l,mid,ceng-t[k*2+1].ce+t[k*2+1].add);
     }
     else
     {
          query_sum(k*2+1,mid+1,r,ceng);
     }
     return ;
}
void updata(int k)//左儿子是否保存
{
     int r=k*2+1;
     int l=k*2;
     //printf("--------
updata k=%lld %lld %lld
",k,l,r);
//     t[k].ce=t[l].ce+t[r].ce;
//     t[k].h=t[l].h+t[r].h;
//     t[k].add=t[l].add+t[r].add;
     if(t[l].ce>t[r].add)
     {
         t[k].ce=t[l].ce+t[r].ce-t[r].add;
         qq_sum=0;
         query_sum(l,t[l].l,t[l].r,t[r].add); 
         t[k].h=t[l].h+t[r].h-qq_sum;
         t[k].add=t[l].add;
         //printf("--t[k].ce=%lld t[k].h=%lld add=%lld
",t[k].ce,t[k].h,t[k].add);
     }
     else
     {
         t[k].h=t[r].h;
         t[k].ce=t[r].ce;
         t[k].add=t[r].add+t[l].add-t[l].ce;
         //printf("t[k].ce=%lld t[k].h=%lld add=%lld
",t[k].ce,t[k].h,t[k].add);      
     }
     //printf("-----------
");
     return ;
}
void build(int k,int l,int r)
{
     t[k].l=l;t[k].r=r;
     if(l==r)
     {
        orz[l]=read();mm[l]=read();
        if(orz[l]==0)
        {
            t[k].ce=1;t[k].h=mm[l];
            //printf("t[%lld].ce=%lld t[%lld].h=%lld
",k,t[k].ce,k,t[k].h);           
        }
        else
        {
            t[k].add=mm[l];
            //printf("t[%lld].add=%lld
",k,t[k].add);           
        }
        return ;
     }
     int mid=(l+r)>>1;
     build(k*2,l,mid);
     build(k*2+1,mid+1,r);
     updata(k);     
}
void F_add(int k,int l,int orzz,int x)
{
     if(t[k].l==t[k].r)
     {
          t[k].add=0;t[k].ce=0;t[k].h=0;
          if(orzz==0)
          {
              t[k].ce=1;
              t[k].h=x;
              //printf("t[%lld].ce=%lld t[%lld].h=%lld
",k,t[k].ce,k,t[k].h);
          }
          else
          {
              t[k].add=x;
              //printf("t[%lld].add=%lld
",k,t[k].add);
          }
          return ;
     }
     int mid=(t[k].l+t[k].r)>>1;
     if(l<=mid)F_add(k*2,l,orzz,x);
     else F_add(k*2+1,l,orzz,x);
     updata(k);
}
signed main()
{
    //freopen("text.in","r",stdin);
   // freopen("kkk.out","w",stdout);
    m=read();q=read();
    //printf("build
");
    build(1ll,1ll,m);
    //printf("t[1].h=%lld
",t[1ll].h);
    for(int i=1;i<=q;++i)
    {
        int x,y,z;
        x=read();y=read();z=read();
        F_add(1,x,y,z);
        printf("%lld
",t[1].h);
    }
}

附上对拍

#include<bits/stdc++.h>
using namespace std;
int main()
{ 
    int c=0;
    while(true)
    {
         system("./pai");//puts("ran work out");
         system("./b");//puts("ac work out");
         system("./kkk");//puts("kx work out");
         if(system("diff -b -B b.out kkk.out"))
         {
            puts("WA");
            return 0;
         }
         cout<<++c<<" ";
         puts("AC");
    }
    return 0;
}
/*
g++ pai.cpp -o pai 
./pai
g++ b.cpp -o b
./b
g++ kkk.cpp -o kkk
./kkk
g++ ran.cpp -o ran
./ran
*/

随机数据生成

#include<bits/stdc++.h>
using namespace std;
int random(int m)
{
    return (long long)rand()*rand()%m;
}
int judge()
{
   int x=rand();
   if(x%2)return 1;
   else return 0;
}
pair<int,int>e[100000];
map<pair<int,int>,bool>h;
int main()
{
    freopen("text.in","w",stdout);
    srand((unsigned)time(0));
    int n=random(7)+1;
    int m=10;
    int q=10;
    printf("%d %d
",m,q);
    for(int i=1;i<=m;++i)
    {
        printf("%d %d
",judge(),random(10)+1);
    }
    for(int i=1;i<=q;++i)
    {
        printf("%d %d %d
",random(m)+1,judge(),random(10)+1);
    }
    return 0;
}