codeforce 1108E Array and Segments 贪心+枚举（线段树维护）

题目链接https://codeforces.com/contest/1108/problem/E2

题意：给你一个长度为n的序列，m个区间，你可以选择若干的区间，对区间内所有数-1。然后设x为修改后的数组中最大元素-最小元素的大小，求最大的x。以及让你给出一个实现的方案，即：需要选择几个区间，选择哪些区间。任意输出一种方案即可。

E1的n和m范围都是300，支持n³的算法，就可以直接枚举。第一层循环i枚举序列最小值的位置，第二层j枚举最大值的位置，再枚举区间，哪些选，哪些不选。如果一个区间包含i且不包含j，那么x就会+1，否则可以不选这个区间。复杂度为n²m。

感觉自己保存区间这块写的有点麻烦，但是赛时想不到其他方法了。。就先AC了再说。。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mp make_pair
#define fi first
#define se second
#define pb push_back
const int maxn=305;
int co[maxn];
pair<int,int> p[maxn];
vector<int>an,tmp;
int main() {
	int n,m,ans=0,ww;
	cin>>n>>m;
	for(int i=1; i<=n; i++)
		cin>>co[i];
	for(int i=1; i<=m; i++) {
		cin>>a>>b;
		p[i]=mp(a,b);
	}
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= n; j++) {
			if(j == i)
				continue;
			int ww = co[j] - co[i], qq = 0;//ww代表最大值和最小值的差
			tmp.clear();
			for(int k=1; k<=m; k++) {
				if(p[k].fi<=i&&p[k].se>=i&&(p[k].fi>j||p[k].se<j)) { //区间包含i且不包含j
					ww++;
					qq++;//选了几个区间
					tmp.pb(k);//用tmp保存这一步选择的区间
				}
			}
			if(ww>ans) {
				an.clear();
				for(int i=0; i<qq; i++)
					an.pb(tmp[i]);//更新an，保存最优解选择的区间
				ans=ww;
				c=qq;//保存最优解选择的区间数
			}
		}
	}
	cout<<ans<<endl;
	cout<<c<<endl;
	for(auto i:an)
		cout<<i<<" ";
}

 

接下来是E2，n的范围来到了1e5，复杂度别说n²m了，连nm都很危险。于是我放弃了优化枚举的想法，重新思考这道题。

首先思考假设i点为最小值，那么包含i点的区间都可以选择。因为就算最大值在这个区间里，选了这个区间也不会对x的大小有影响，因为最大最小值都-1。而如果最大值不在这个区间里，那么你选择这个区间后，最小值-1，x就会+1。而如果一个区间不包含最小值的位置，显然，尽量不选。

那么我们可以利用差分数组的思想，保存每一个区间的起始位置和结束位置，然后从1开始枚举最小值位置。那么现在假设1是最小值的位置，我们就应该把包含1的区间全部选中，那么显然，只有以1开始的区间是包含1的，那么我们直接修改数组，把这些区间全部-1。然后再去寻找数组中的最大值，就可以算出1是最小值时的x。

然后枚举到2，因为1开始的区间已经在数组里改过了，那么我们只需要把2开始的区间再全部选中并且-1，那么所有包含2的区间不就都被选中了吗？这时候再寻找数组中的最大值，就可以算出2是最小值时的x。如果有一个区间是从1到1怎么办？枚举到2的时候这个区间是不应该被选的（因为不包含2），所以在判断完位置1之后，应该把所有以1结束的区间重新“退选”，即把这些区间再+1，恢复到没选的状态。这样就保证在枚举i位置为最小值时，当前选中的区间都包含i，没有其他无关区间。

区间修改+区间查询最大值，用线段树维护即可。对于每个区间，只有选择和退选的操作，也就是只修改了2*m次的区间。对于每个位置的i都需要查询最大值。复杂度为(n+2m)*logn，由于m很小，复杂度约为nlogn。

找方案时，根据一开始的思想，只要记录了最小值的位置i，那么把所有包含i的区间都选上即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mp make_pair
#define fi first
#define se second
#define pb push_back
const int maxn=1e5+5;
const ll mod=998244353;
const int inf=0x3f3f3f3f;
ll co[maxn],dqs[maxn],f[maxn],val[maxn];
pair<int,int> p[maxn];
vector<int>tp,ed[maxn],st[maxn];
class segtree{public:
#define nd node[now]
#define ndl node[now<<1]
#define ndr node[now<<1|1]
	struct segnode{
		int l,r,val,tag;//表示[l,r]的最大值为val,并且标记为tag
		void update(int x){val+=x,tag+=x;}
	};
	vector<segnode> node;
	int n;
	int uu(int now){return nd.val;
	}
	void pushup(int now){nd.val=max(ndl.val,ndr.val);}
	//由被修改的儿子去更新父节点信息
	void pushdown(int now){
		if(nd.tag){
			ndl.update(nd.tag);
			ndr.update(nd.tag);
			nd.tag=0;
		}
	}
	void maketree(int s,int t,int now=1){//构造树
		nd={s,t,0};
		if(s==t) {
			cin>>nd.val;
			return ;
		}
		maketree(s,(s+t)>>1,now<<1);
		maketree(((s+t)>>1)+1,t,now<<1|1);
		pushup(now);
	}
	void update(int s,int t,int x,int now=1){ 
		if(s<=nd.l&&t>=nd.r) {nd.update(x);return ;}//递归到单点结束
		pushdown(now);
		if(s<ndr.l) update(s,t,x,now<<1);
		if(t>ndl.r) update(s,t,x,now<<1|1);
		pushup(now);
	}
	int query(int s,int t,int now=1){//区间查询
		if(s<=nd.l&&t>=nd.r) return nd.val;//递归到子区间被询问覆盖
		pushdown(now);
		int ret=-1e9;
		if(s<ndr.l) ret=query(s,t,now<<1);
		if(t>ndl.r) ret=max(query(s,t,now<<1|1),ret);
		return ret;
	}
	segtree(int nn){
		n=nn;
		node.resize(n<<2|3);
		maketree(1,n);
	}
};
int main(){
	int n,m,k,t,a,b,c,d,ww,maxx=-1,ans=0;
	cin>>n>>m;
	segtree tree(n);
	for(int i=1;i<=m;i++){
		scanf("%d%d",&a,&b);
		p[i]=mp(a,b);
		st[a].pb(b);//存以a为起点的区间 
		ed[b].pb(a);//存以b为终点的区间 
	}
	for(int i=1;i<=n;i++){//枚举最小值的位置 
		for(int j=0;j<st[i].size();j++)
			tree.update(i,st[i][j],-1);//把包含最小值的区间全部-1 
		int tmp=tree.query(1,n)-tree.query(i,i);
		if(tmp>maxx){
			a=i;//记录最小值的位置 
			maxx=tmp;//记录最小值 
		}
		for(int j=0;j<ed[i].size();j++)
			tree.update(ed[i][j],i,1);//此点判断完毕，把以此点为终点的区间改回来 
	}
	printf("%d\n",maxx);
	for(int i=1;i<=m;i++){
		if(p[i].first<=a&&p[i].se>=a){
			ans++;
			tp.pb(i);
		}
	}
	printf("%d\n",ans);
	for(auto i:tp)
		printf("%d ",i);	
}