线段树判一段里面有几段不同的颜色以及更新操作

#include <bits/stdc++.h>
using namespace std;
const int MAXN =3e5+100;
#define L(x) (x<<1)
#define R(x) (x<<1|1)

int a[MAXN];
typedef long long ll;
struct node
{
int l,r;
int num;
ll tag,lc,rc;
}segtree[MAXN<<2];

void push_up(int rt)
{
segtree[rt].lc=segtree[L(rt)].lc;segtree[rt].rc=segtree[R(rt)].rc;
int ans=segtree[L(rt)].num+segtree[R(rt)].num;
if(segtree[L(rt)].rc==segtree[R(rt)].lc)
ans--;
segtree[rt].num=ans;
}

void push_down(int rt)
{
if(segtree[rt].tag){
segtree[L(rt)].tag+=segtree[rt].tag;
segtree[R(rt)].tag+=segtree[rt].tag;
segtree[L(rt)].lc+=segtree[rt].tag;
segtree[L(rt)].rc+=segtree[rt].tag;
segtree[R(rt)].lc+=segtree[rt].tag;
segtree[R(rt)].rc+=segtree[rt].tag;
segtree[rt].tag=0;
}
}

void build(int rt,int l,int r)
{
segtree[rt].l=l;segtree[rt].r=r;segtree[rt].num=0;
if(l==r)
{
segtree[rt].num=1,segtree[rt].lc=segtree[rt].rc=1ll*a[l];
return ;
}
int mid=(l+r)>>1;
build(L(rt),l,mid);build(R(rt),mid+1,r);
push_up(rt);
}

void update(int l,int r,ll x,int rt)
{
if(l<=segtree[rt].l&&r>=segtree[rt].r)
{
segtree[rt].tag+=x;
segtree[rt].lc+=x;
segtree[rt].rc+=x;
return ;
}
push_down(rt);
int mid=(segtree[rt].l+segtree[rt].r)>>1;
if(l<=mid) update(l,r,x,L(rt));
if(r>mid) update(l,r,x,R(rt));
push_up(rt);
}

int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,1,n);
int q;
scanf("%d",&q);
while(q--)
{
int l,r;
ll d;
scanf("%d%d%lld",&l,&r,&d);
update(l,r,d,1);
printf("%d\n",segtree[1].num );
}
}