首页 > 代码库 > ACdream 1101 瑶瑶想要玩滑梯

ACdream 1101 瑶瑶想要玩滑梯

没想到线段树的基本用法这么长时间没写了还没有忘,1A的感觉还是很爽的。

题目大意:

中文题,点此查看题目。


解题思路:

线段树的区间更新与查询。

lazy标记的使用。

当需要返回区间多个值时可以使用引用参数。



下面是代码:

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <math.h>
#include <stdlib.h>

using namespace std;

int min(int a,int b)
{
    if(a>b)a=b;
    return a;
}
int max(int a,int b)
{
    if(a<b)a=b;
    return a;
}
int n,m,l,r,x;
struct node1
{
    int num,l,r,lcnt,rcnt,mcnt;
} node[100000<<2];
char s[3];
void PushUp(int l,int r,int m,int tr)
{
    node[tr].num=-1;
    node[tr].l=node[tr<<1].l;
    node[tr].r=node[tr<<1|1].r;
    node[tr].lcnt=node[tr<<1].lcnt;
    if(node[tr<<1].lcnt==m-l+1&&node[tr<<1].r<node[tr<<1|1].l)
    {
        node[tr].lcnt+=node[tr<<1|1].lcnt;
    }
    node[tr].rcnt=node[tr<<1|1].rcnt;
    if(node[tr<<1|1].rcnt==r-m&&node[tr<<1].r<node[tr<<1|1].l)
    {
        node[tr].rcnt+=node[tr<<1].rcnt;
    }
    if(node[tr<<1].r<node[tr<<1|1].l)
    {
        node[tr].mcnt=node[tr<<1].rcnt+node[tr<<1|1].lcnt;
    }
    else node[tr].mcnt=0;
    node[tr].mcnt=max(node[tr<<1].mcnt,max(node[tr<<1|1].mcnt,max(node[tr].lcnt,max(node[tr].rcnt,node[tr].mcnt))));
}
void build(int l,int r,int tr)
{
    if(l==r)
    {
        scanf("%d",&node[tr].num);
        node[tr].l=node[tr].num;
        node[tr].r=node[tr].num;
        node[tr].lcnt=1;
        node[tr].rcnt=1;
        node[tr].mcnt=1;
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,tr<<1);
    build(m+1,r,tr<<1|1);
    PushUp(l,r,m,tr);
}
void Pushdown(int l,int r,int tr)
{
    if(node[tr].num!=-1)
    {
        node[tr<<1]=node[tr];
        node[tr<<1|1]=node[tr];
        node[tr].num=-1;
    }
}
void Update(int L,int R,int num,int l,int r,int tr)
{
    if(L<=l&&r<=R)
    {
        if(node[tr].num==-1)node[tr].num=0;
        node[tr].num+=num;
        node[tr].l=num;
        node[tr].r=num;
        node[tr].lcnt=1;
        node[tr].rcnt=1;
        node[tr].mcnt=1;
        return ;
    }
    if(l==r)return;
    Pushdown(l,r,tr);
    int m=(l+r)>>1;
    if(m>=L)Update(L,R,num,l,m,tr<<1);
    if(m<R)Update(L,R,num,m+1,r,tr<<1|1);
    PushUp(l,r,m,tr);
}
void copy1(int a,int b,int c,int d,int e,int &a1,int &b1,int &c1,int &d1,int &e1)
{
    a1=a;
    b1=b;
    c1=c;
    d1=d;
    e1=e;
}
void query(int L,int R,int l,int r,int tr,int &lcnt,int &lnum,int &rcnt,int &rnum,int &mnum)
{
    if(L<=l&&r<=R)
    {
        mnum=node[tr].mcnt;
        lcnt=node[tr].lcnt;
        lnum=node[tr].l;
        rcnt=node[tr].rcnt;
        rnum=node[tr].r;
        return;
    }
    if(l==r)return;
    int m=(l+r)>>1,a[2]={0},b[2]={0},c[2]={0},d[2]={0},e[2]={0};
    Pushdown(l,r,tr);
    if(m>=L)
    {
        query(L,R,l,m,tr<<1,a[0],b[0],c[0],d[0],e[0]);
    }
    if(m<R)
    {
        query(L,R,m+1,r,tr<<1|1,a[1],b[1],c[1],d[1],e[1]);
    }
    if(c[0]!=0)
    {
        if(a[1]!=0)
        {
            mnum=max(e[0],e[1]);
            lcnt=a[0];
            lnum=b[0];
            rcnt=c[1];
            rnum=d[1];
            if(d[0]<b[1])
            {
                mnum=max(mnum,c[0]+a[1]);
                if(a[0]==e[0]&&a[0]==m-l+1)
                {
                    lcnt+=a[1];
                }
                if(c[1]==e[1]&&c[1]==r-m)
                {
                    rcnt+=c[0];
                }
            }
        }
        else
        {
            copy1(a[0],b[0],c[0],d[0],e[0],lcnt,lnum,rcnt,rnum,mnum);
        }
    }
    else
    {
        copy1(a[1],b[1],c[1],d[1],e[1],lcnt,lnum,rcnt,rnum,mnum);
    }
    PushUp(l,r,m,tr);
    return ;
}
int main()
{
    scanf("%d%d",&n,&m);
    build(1,n,1);
    int temp[5];
    for(int i=0; i<m; i++)
    {
        scanf("%s",s);
        if(s[0]=='Q')
        {
            scanf("%d%d",&l,&r);
            query(l,r,1,n,1,temp[0],temp[1],temp[2],temp[3],temp[4]);
            printf("%d\n",max(temp[0],max(temp[2],temp[4])));
        }
        else if(s[0]=='U')
        {
            scanf("%d%d%d",&l,&r,&x);
            Update(l,r,x,1,n,1);
        }
    }
    return 0;
}