2020牛客暑期多校训练营(第四场)E Eliminate++ —— 线段树优化

This way

题意:

给你一个长度为n的排列,你每次可以取一个长度为3的子串,然后消掉最大和最小的数,直到最终剩下一个数,问你每个数是否可能是那个数。

题解:

那么我们肯定是枚举每个数,然后查看有多少数大于他,有多少数小于它。如果这些是相同的,那么每次就可以找连续三个数使得有一个大于它,一个小于它,然后直到消到最后。
那么如果不相同,我们就可以消掉连续三个大于它或者连续三个小于它中的两个。接下来用1表示大于当前数,0表示小于。
但是不能每次只看连续的三个相同数,比如11011这个串,就可以先消掉一个01,然后再消掉两个1.
然后1和0数量的情况要分开做。
接下来讨论1大于0的情况
并且这是满足贪心的,从左到右做的时候,如果当前手上有大于等于三个1,那么必然可以消掉两个1,否则如果还有1的话,那么就要和这个0抵消掉。因此线段树维护的东西不满足交换律。
那么用两个数组维护我们需要的值:
num[i][j]表示在根为i的时候,如果从左边传进来j个1,可以消掉的连续3个1的次数
res[i][j]表示在根为i的时候,如果从左边传进来j个1,到最后剩下的1的个数。
很明显j的值是0~2
那么初始化按照上面的逻辑就是这个样子:

		for(int i=0;i<3;i++)
            num[root][i]=i>=2,res[root][i]=i%2+1;
将当前点置为0的时候:
		for(int i=0;i<3;i++)
            num[root][i]=0,res[root][i]=i-1;
        res[root][0]=0;

表示当前点不能消掉1,然后往右传的时候1的数量会减少1。
之后query的时候需要从左到右
在比赛的时候没有想到贪心,所以没做出来,感觉还是蛮可惜的,因为并不是一道非常难的题目
由于是参考别人的解法,这个时间复杂度还是比较可以的:

在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int num[N*4][3],res[N*4][3];
void push_up(int root){
    for(int i=0;i<3;i++){
        num[root][i]=num[root<<1][i]+num[root<<1|1][res[root<<1][i]];
        res[root][i]=res[root<<1|1][res[root<<1][i]];
    }
}
void build(int l,int r,int root,int p){
    if(l==r){
        for(int i=0;i<3;i++)
            num[root][i]=i>=2,res[root][i]=i%2+1;
        return ;
    }
    int mid=l+r>>1;
    build(l,mid,root<<1,p);
    build(mid+1,r,root<<1|1,p);
    push_up(root);
}
void update(int l,int r,int root,int p){
    if(l==r){
        for(int i=0;i<3;i++)
            num[root][i]=0,res[root][i]=i-1;
        res[root][0]=0;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=p)
        update(l,mid,root<<1,p);
    else
        update(mid+1,r,root<<1|1,p);
    push_up(root);
}
void query(int l,int r,int root,int ql,int qr,int &rs,int &ans){
    if(l>=ql&&r<=qr){
        ans+=num[root][rs];
        rs=res[root][rs];
        return ;
    }
    int mid=l+r>>1;
    if(mid>=ql)
        query(l,mid,root<<1,ql,qr,rs,ans);
    if(mid<qr)
        query(mid+1,r,root<<1|1,ql,qr,rs,ans);
}
int a[N],p[N],v[N];
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]),p[a[i]]=i;
        int mid=(n+1)/2;
        build(1,n,1,p[1]);
        int ans=0;
        for(int i=1;i<mid;i++){
            int rs=0,l=1,r=p[i]-1;
            ans=0;
            update(1,n,1,p[i]);
            if(l<=r)
                query(1,n,1,l,r,rs,ans);
            rs=0;
            l=p[i]+1,r=n;
            if(l<=r)
                query(1,n,1,l,r,rs,ans);
            if(ans*2+i-1>=n-i)v[p[i]]=1;
            else v[p[i]]=0;
        }
        v[p[mid]]=1;
        build(1,n,1,p[n]);
        for(int i=n;i>mid;i--){
            int rs=0,l=1,r=p[i]-1;
            ans=0;
            update(1,n,1,p[i]);
            if(l<=r)
                query(1,n,1,l,r,rs,ans);
            rs=0;
            l=p[i]+1,r=n;
            if(l<=r)
                query(1,n,1,l,r,rs,ans);
            if(ans*2+(n-i)>=i-1)v[p[i]]=1;
            else v[p[i]]=0;
        }
        for(int i=1;i<=n;i++)
            printf("%d",v[i]);
        printf("\n");
    }
    return 0;
}
/*
1
3
2 3 1
*/

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/107471515