E. Painting the Fence——东二(双端队列,区间线段树)

版权声明:转载记得标明出处哦~ https://blog.csdn.net/weixin_43890047/article/details/89046036

题目地址:http://codeforces.com/gym/101911/problem/E

这是一道好的算法组合题,涉及线段树以及双端队列。

做完这道题,先写收获:
1、差分树状数组(用于区间修改)不能处理区间统一换成数值v的问题(对我而言),只能简单的用作区间加减乘除运算。
2、在维护线段树区间的元素的数量的时候,以及其他统计运算的时候,最优思想不应该是将统计元素数量(以及其他需求)整合到树状数组中,而应优先考虑分开思想,即将另一个需求分开来实现。整合思想和数据结构二分思想。
3、接2,在统计区间元素数量的时候,双向队列,set(最好不要再构建树了,思考后再使用),栈,map,看看那个容器能符合题目的要求。
4、如果线段树只用维护树的最低端,即不需要维护区间和、最大值等区间信息的时候,lazy标记查询到l==r的时候,col[p]就是树的最低端,不需要在构建s[p]的线段树了

AC代码

#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 3e5+50;
int col[4*MAX_N]={0};
int a[MAX_N] = {0};
int n;
int m;
deque<int> dq[MAX_N];
void down(int p)
{
    if(col[p]){
        col[2*p] = col[p];
        col[2*p+1] = col[p];
        col[p] = 0;
    }
}

void build(int p,int l,int r)
{
    if(l==r){
        col[p] = a[l];
        return ;
    }
    int mid = (l+r)/2;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
}

void change(int p,int l,int r,int x,int y,int v)
{
    if(x<=l&&r<=y){
        col[p] = v;
        return ;
    }
    down(p);
    int mid = (l+r)/2;
    if(x<=mid){
        change(p*2,l,mid,x,y,v);
    }
    if(y>mid){
        change(p*2+1,mid+1,r,x,y,v);
    }
}

int query(int p,int l,int r,int ind)
{
    if(l==r){
        return col[p];
    }
    down(p);
    int mid = (l+r)/2;
    if(ind<=mid){
        return query(p*2,l,mid,ind);
    }
    else{
        return query(p*2+1,mid+1,r,ind);
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",a+i);
        int k = a[i];
        dq[k].push_back(i);     ///一开始就排好了
    }
    build(1,1,n);
    scanf("%d",&m);
    while(m--){
        int x;
        scanf("%d",&x);
        if(dq[x].size()<=1) continue;
        int s = dq[x].front();
        int t = dq[x].back();
        while(query(1,1,n,s)!=x&&!dq[x].empty()){
            dq[x].pop_front();
            if(!dq[x].empty()){
                s = dq[x].front();
            }
        }
        while(query(1,1,n,t)!=x&&!dq[x].empty()){
            dq[x].pop_back();
            if(!dq[x].empty()){
                t = dq[x].back();
            }
        }
        if(dq[x].size()>1){
            change(1,1,n,s,t,x);
        }
    }
    int flag = 1;
    for(int i=1;i<=n;i++){
        if(!flag) printf(" ");
        flag = 0;
        printf("%d",query(1,1,n,i));
    }
    printf("\n");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43890047/article/details/89046036
今日推荐