HDU - 1754 I Hate It(线段树求RMQ)

给你一个区间,求这个区间的最大值,并且要进行点更新,跟区间更新没什么区别的,不过涉及到区间更新,属于动态RMQ,静态RMQ有更简单的做法(https://blog.csdn.net/qq_38367681/article/details/81143373),本题代码如下:

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>

using namespace std;

const int MAXN=200010;
int a[MAXN],ans[MAXN<<2],lazy[MAXN<<2];

void PushUp(int rt)
{
    ans[rt]=max(ans[rt<<1],ans[rt<<1|1]);
}

void Build(int l,int r,int rt)
{
    if (l==r)
    {
        ans[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    Build(l,mid,rt<<1);
    Build(mid+1,r,rt<<1|1);
    PushUp(rt);
}

void PushDown(int rt,int ln,int rn)//ln表示左子树元素结点个数,rn表示右子树结点个数
{
    if (lazy[rt])
    {
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        ans[rt<<1]+=lazy[rt]*ln;
        ans[rt<<1|1]+=lazy[rt]*rn;
        lazy[rt]=0;
    }
}

void Add(int L,int C,int l,int r,int rt)
{
    if (l==r)
    {
        ans[rt]=C;
        return;
    }
    int mid=(l+r)>>1;
    //PushDown(rt,mid-l+1,r-mid); 若既有点更新又有区间更新,需要这句话
    if (L<=mid)
        Add(L,C,l,mid,rt<<1);
    else
        Add(L,C,mid+1,r,rt<<1|1);
    PushUp(rt);
}

void Update(int L,int R,int C,int l,int r,int rt)
{
    if (L<=l&&r<=R)
    {
        ans[rt]+=C*(r-l+1);
        lazy[rt]+=C;
        return;
    }
    int mid=(l+r)>>1;
    PushDown(rt,mid-l+1,r-mid);
    if (L<=mid) Update(L,R,C,l,mid,rt<<1);
    if (R>mid) Update(L,R,C,mid+1,r,rt<<1|1);
    PushUp(rt);
}

int Query(int L,int R,int l,int r,int rt)
{
    if (L<=l&&r<=R)
        return ans[rt];
    int mid=(l+r)>>1;
    //PushDown(rt,mid-l+1,r-mid);//若更新只有点更新,不需要这句
    int ANS=0;
    if (L<=mid) ANS=max(ANS, Query(L,R,l,mid,rt<<1));
    if (R>mid)  ANS=max(ANS, Query(L,R,mid+1,r,rt<<1|1));
    return ANS;
}

int main()
{
    int T, n, p, q, t = 1, m;
    string s;
    while(cin >> n >> m) {
        memset(lazy, 0, sizeof(lazy));
        memset(ans, 0, sizeof(ans));
        for(int i = 1; i <= n; i++) {
            cin >> a[i];
        }
        Build(1, n, 1);
        while(m--) {
            cin >> s >> p >> q;
            if(s[0] == 'U') {
                Add(p, q, 1, n, 1);
            }
            else {
                printf("%d\n", Query(p, q, 1, n, 1));
            }
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38367681/article/details/81270559
今日推荐