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

代码
一道题目看错题意写出来的。
有更新操作，就是往某个区间每一个数加值，再看所有区间有几段不同的

样例
Input
5
5 5 5 5 5
3
1 3 2
2 2 1
4 4 1
Output
2
4
5

#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 );
    }
}