牛客练习赛56 D.小翔和泰拉瑞亚/Codeforces Round #535 (Div. 3).Array and Segments (Hard version)(线段树/扫描线思想)

题目

小翔爱玩泰拉瑞亚 。一天，他碰到了一幅地图。

这幅地图可以分为n列，第i列的高度为Hi，他认为这个地图不好看，决定对它进行改造。

小翔又学会了m个魔法，实施第i个魔法可以使地图的第Li列到第Ri列每一列的高度减少Wi，

每个魔法只能实施一次，魔法的区间可能相交或包含。

 

小翔认为，一幅地图中最高的一列与最低的一列的高度差越大，这幅地图就越美观。

小翔可以选择m个魔法中的任意一些魔法来实施，使得地图尽量美观。

但是他不知道该如何选择魔法，于是他找到了你。请你求出所有可行方案中，高度差的最大值。

对于100%的数据，满足1≤n,m≤2e5,-1e9≤Hi≤1e9,1≤Wi≤1e9,1≤Li≤Ri≤n。

题解

考虑枚举最大列i，即包含最大列i的魔法不使用，其余列的魔法均使用，

初始情况，把所有魔法都使用，然后考虑撤回第i列的操作，然后询问a[i]-区间[1,n]的最小值

开始想着类似时间分治离线建树，dfs进入这个点就把这个区间的魔法都撤回，离开时再使用所有魔法，

但后来发现自己想多了，一个线性结构非要弄成树形的干什么……

先使用所有魔法，在从左到右扫区间，li放入以l为左端点的区间，ri放入以r为右端点的区间

进入时使用撤回操作，询问最值，离开时取消掉撤回操作，O(mlogn)

后来发现，这个东西不就是扫描线吗orz……

代码

#include<bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
#define lson p<<1,l,mid
#define rson p<<1|1,mid+1,r
using namespace std;
typedef long long ll;
const int N=2e5+10;
typedef pair<ll,ll> P;
int n,m,l,r;
ll up,down,w,a[N],ans;
vector<ll>L[N],R[N];
struct node{
    int l,r,w;
    node(){}
    node(int a,int b,int c):l(a),r(b),w(c){
    }
}e[N];
struct segtree{
	int n;
	struct node{int l,r;ll v,cov;}e[N<<2];
	#define l(p) e[p].l
	#define r(p) e[p].r
	#define v(p) e[p].v
	#define c(p) e[p].cov
	void up(int p){v(p)=min(v(p<<1),v(p<<1|1));}
	void bld(int p,int l,int r){
		l(p)=l;r(p)=r;c(p)=0;
		if(l==r){v(p)=a[l];return;}
		int mid=l+r>>1;
		bld(p<<1,l,mid);bld(p<<1|1,mid+1,r);
		up(p);
	}
	void psd(int p){
	    if(c(p)){
            v(p<<1)+=c(p);c(p<<1)+=c(p);
            v(p<<1|1)+=c(p);c(p<<1|1)+=c(p);
            c(p)=0;
	    }
	}
	void init(int _n){n=_n;bld(1,1,n);}
	void chg(int p,int ql,int qr,ll w){
		if(ql<=l(p)&&r(p)<=qr){
            c(p)+=w;
            v(p)+=w;
            return;
		}
		psd(p);
		int mid=l(p)+r(p)>>1;
		if(ql<=mid)chg(p<<1,ql,qr,w);
		if(qr>mid)chg(p<<1|1,ql,qr,w);
		up(p);
	}
	ll ask(int p,int ql,int qr){
		if(ql<=l(p)&&r(p)<=qr)return v(p);
		int mid=l(p)+r(p)>>1;ll res=1e18;
		psd(p);
		if(ql<=mid)res=min(res,ask(p<<1,ql,qr));
		if(qr>mid)res=min(res,ask(p<<1|1,ql,qr));
		return res;
	}
}seg;
int main(){
    scanf("%d%d",&n,&m);
    up=-1e9;down=1e9;
    for(int i=1;i<=n;++i){
        scanf("%lld",&a[i]);
        up=max(up,a[i]);
        down=min(down,a[i]);
    }
    ans=max(ans,up-down);
    seg.init(n);
    for(int i=1;i<=m;++i){
        scanf("%d%d%lld",&l,&r,&w);
        seg.chg(1,l,r,-w);
        e[i]=node(l,r,w);
        L[l].push_back(i);
        R[r].push_back(i);
    }
    for(int j=1;j<=n;++j){
        for(int i:L[j]){
            seg.chg(1,e[i].l,e[i].r,e[i].w);
        }
        ans=max(ans,seg.ask(1,j,j)-seg.ask(1,1,n));
        for(int i:R[j]){
            seg.chg(1,e[i].l,e[i].r,-e[i].w);
        }
    }
    printf("%lld\n",ans);
    return 0;
}

Bouns

Codeforces Round #535 (Div. 3)

E2. Array and Segments (Hard version)

n<=1e5，m<=300，-1e6<=ai<=1e6，w=1，但要求输出魔法方案

m<=300，set动态维护一下即可

代码

#include<bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
#define lson p<<1,l,mid
#define rson p<<1|1,mid+1,r
using namespace std;
typedef long long ll;
const int N=1e5+10;
typedef pair<ll,ll> P;
int n,m,l,r;
ll up,down,w,a[N],ans;
vector<ll>L[N],R[N];
set<int>now,res;
struct node{
    int l,r,w;
    node(){}
    node(int a,int b,int c):l(a),r(b),w(c){
    }
}e[N];
struct segtree{
	int n;
	struct node{int l,r;ll v,cov;}e[N<<2];
	#define l(p) e[p].l
	#define r(p) e[p].r
	#define v(p) e[p].v
	#define c(p) e[p].cov
	void up(int p){v(p)=min(v(p<<1),v(p<<1|1));}
	void bld(int p,int l,int r){
		l(p)=l;r(p)=r;c(p)=0;
		if(l==r){v(p)=a[l];return;}
		int mid=l+r>>1;
		bld(p<<1,l,mid);bld(p<<1|1,mid+1,r);
		up(p);
	}
	void psd(int p){
	    if(c(p)){
            v(p<<1)+=c(p);c(p<<1)+=c(p);
            v(p<<1|1)+=c(p);c(p<<1|1)+=c(p);
            c(p)=0;
	    }
	}
	void init(int _n){n=_n;bld(1,1,n);}
	void chg(int p,int ql,int qr,ll w){
		if(ql<=l(p)&&r(p)<=qr){
            c(p)+=w;
            v(p)+=w;
            return;
		}
		psd(p);
		int mid=l(p)+r(p)>>1;
		if(ql<=mid)chg(p<<1,ql,qr,w);
		if(qr>mid)chg(p<<1|1,ql,qr,w);
		up(p);
	}
	ll ask(int p,int ql,int qr){
		if(ql<=l(p)&&r(p)<=qr)return v(p);
		int mid=l(p)+r(p)>>1;ll res=1e18;
		psd(p);
		if(ql<=mid)res=min(res,ask(p<<1,ql,qr));
		if(qr>mid)res=min(res,ask(p<<1|1,ql,qr));
		return res;
	}
}seg;
bool upd(ll &x,ll y){
    return x<y?x=y:0;
}
int main(){
    scanf("%d%d",&n,&m);
    up=-1e9;down=1e9;
    for(int i=1;i<=n;++i){
        scanf("%lld",&a[i]);
        up=max(up,a[i]);
        down=min(down,a[i]);
    }
    ans=max(ans,up-down);
    seg.init(n);
    for(int i=1;i<=m;++i){
        scanf("%d%d",&l,&r);w=1;
        seg.chg(1,l,r,-w);
        now.insert(i);
        e[i]=node(l,r,w);
        L[l].push_back(i);
        R[r].push_back(i);
    }
    for(int j=1;j<=n;++j){
        for(int i:L[j]){
            seg.chg(1,e[i].l,e[i].r,e[i].w);
            now.erase(i);
        }
        if(upd(ans,seg.ask(1,j,j)-seg.ask(1,1,n))){
            res=now;
        }
        for(int i:R[j]){
            seg.chg(1,e[i].l,e[i].r,-e[i].w);
            now.insert(i);
        }
    }
    printf("%lld\n",ans);
    printf("%d\n",(int)res.size());
    for(int x:res){
        printf("%d ",x);
    }
    return 0;
}