题目:Problem - 1108E2 - Codeforces
大意:给你一个长度为n的数组a,还有m个区间,让你选任意几个区间,让这个几个区间的数都-1
一顿操作后让a总和最小,打印出任一方案(区间标号)
主要就是枚举:首先谁当最小值跟最大值是不确定的,选哪个区间也是不确定的,直接枚举(n3)可以过easy版本,n为10e5后,需要简化一下枚举,经讨论:max跟min在同一更新区间里时,不会影响结果,只有min在更新区间里效果更优,仅有max在更新区间里效果更差,所以枚举过更新区间时,发现min不在其中则没有必要枚举,第一步优化可以边枚举min边枚举更新区间(发现min进入更新区间时就让这个区间-1,出了某个更新新区间时,就让这个区间恢复+1,防止降低区间外的max)目的就是找尽量多的区间降低min的值(贪心),第二步优化就可以用线段树进行区间加减,和记录区间最值整体复杂度(nlog(n))
#include<stdio.h>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
int inf=0x3f3f3f3f,a[1001000],tree[1991919],lz[1001008],s[1004001],e[1400103];
vector<int > vsl[1000100],vsr[1000100],vs;
void build(int node,int l,int r)
{
int mid=(l+r)/2;
if(l==r)
{
tree[node]=a[l];
return;
}
build(node*2,l,mid);
build(node*2+1,mid+1,r);
tree[node]=max(tree[node*2+1],tree[node*2]);
}
void push_down(int node)
{
if(lz[node])
{
lz[node*2+1]+=lz[node];/*注意是加*/
lz[node*2]+=lz[node];
tree[node*2]+=lz[node];
tree[node*2+1]+=lz[node];
}
lz[node]=0;
}
void update(int node,int l,int r,int L,int R,int x)
{
int mid=(l+r)/2;
if(l>=L&&r<=R)
{
lz[node]+=x;
tree[node]+=x;
return;
}
push_down(node);
if(L<=mid) update(node*2,l,mid,L,R,x);
if(R>mid) update(node*2+1,mid+1,r,L,R,x);
tree[node]=max(tree[node*2+1],tree[node*2]);
}
int query(int node,int l,int r,int L,int R)
{
int mid=(l+r)/2,ans1,ans2;
if(l>=L&&r<=R)
{
return tree[node];
}
if(l>R||r<L) return (-inf);
push_down(node);
ans1=query(node*2,l,mid,L,R);
ans2=query(node*2+1,mid+1,r,L,R);
return max(ans1,ans2);
}
int main()
{
int i,j,k,l,m,n,r,max1=-inf,pos,ans;
scanf("%d%d",&n,&m);
for(i=1; i<=n; i++)
scanf("%d",&a[i]);
build(1,1,n);
for(i=1; i<=m; i++)
{
scanf("%d %d",&s[i],&e[i]);
vsl[s[i]].push_back(e[i]);/*记录这个起点结束的位置*/
vsr[e[i]].push_back(s[i]);
}
for(i=1; i<=n; i++)
{
for(j=0; j<vsl[i].size(); j++)/*找找min都进入了哪些区间*/
{
update(1,1,n,i,vsl[i][j],-1);
}
ans=tree[1]-query(1,1,n,i,i);
if(max1<ans)
{
pos=i;/*更新区间不必记录在外面单独找就行*/
max1=ans;
}
for(j=0; j<vsr[i].size() ; j++)/*找找min+1都出了哪些区间*/
{
update(1,1,n,vsr[i][j],i,1);
}
}
for(i=1; i<=m; i++)
{
if(s[i]<=pos&&pos<=e[i])
{
vs.push_back(i);
}
}
printf("%d\n%d\n",max1,vs.size());
for(i=0; i<vs.size(); i++)
{
printf("%d ",vs[i]);
}
return 0;
}