线段树莫名写挂了,求调
查看原帖
线段树莫名写挂了,求调
848964
hzoi_Shadow楼主2023/7/16 22:34
#include<bits/stdc++.h>
using namespace std;
int a[4000001],n,m,i,l,r;
char pd;
struct student
{
    int l,r,num,maxx=0;
}tree[4000001];
int lson(int x)
{
    return x*2;
}
int rson(int x)
{
    return x*2+1;
}
void pushup(int rt)
{
    tree[rt].maxx=max(tree[lson(rt)].maxx,tree[rson(rt)].maxx);
}
void build(int rt,int l,int r)
{
    tree[rt].l=l;
    tree[rt].r=r;
    if(l==r)
    {
        tree[rt].maxx=a[l];
        return;
    }
    int mid=(l+r)/2;
    build(lson(rt),l,mid);
    build(rson(rt),mid+1,r);
    pushup(rt);
}
int add(int rt)
{
    if(tree[rt].l==tree[rt].r)
    {
        tree[rt].num=a[tree[rt].l];
    }
    else
    {
        tree[rt].num=add(lson(rt))+add(rson(rt));
    }
    return tree[rt].num;
}
void update(int rt,int pos,int val)
{
    tree[rt].num+=val;
    if(tree[rt].l==tree[rt].r)
    {
        tree[rt].maxx=max(tree[rt].num,tree[rt].maxx);
        return;
    }
    int mid=(tree[rt].l+tree[rt].r)/2;
    if(pos<=tree[lson(rt)].r)
    {
        update(lson(rt),pos,val);
    }
    if(pos>=tree[rson(rt)].l)
    {
        update(rson(rt),pos,val);
    }
}
int query(int rt,int l ,int r)
{
    if(l==tree[rt].l&&tree[rt].r==r)
    {
        return tree[rt].maxx;
    }
    int mid=(tree[rt].l+tree[rt].r)/2;
    if(r<=mid)
    {
        return query(lson(rt),l,r);
    }
    if(mid<l)
    {
        return query(rson(rt),l,r);
    }
    return max(query(lson(rt),l,mid),query(rson(rt),mid+1,r));
}
int main()
{
    cin>>n>>m;
    for(i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    build(1,1,n);
    add(1);
    for(i=1;i<=m;i++)
    {
        cin>>pd>>l>>r;
        if(pd=='U')
        {
            if(a[l]<r)
            {
                a[l]=r;
                update(1,l,r-a[l]);
            }
        }
        else
        {
            cout<<query(1,l,r)<<" "<<endl;
        }
    }
    return 0;
}
2023/7/16 22:34
加载中...