Codeforces 1108E2 Array and Segments (Hard version)（差分+思维）

题目链接：Array and Segments (Hard version)

题意：给定一个长度为n的序列，m个区间，从m个区间内选择一些区间内的数都减一，使得整个序列的最大值减最小值最大。

题解：利用差分的思想，并且考虑到m比较小，遍历一遍序列，当前点遇到需要改变的时候进行操作，同时更新答案。

#include <set>
#include <map>
#include <queue>
#include <deque>
#include <stack>
#include <cmath>
#include <cstdio>
#include <vector>
#include <string>
#include <cstring>
#include <fstream>
#include <iostream>
#include <algorithm>
using namespace std;

#define eps 1e-8
#define pb push_back
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define clr(a,b) memset(a,b,sizeof(a)
#define bugc(_) cerr << (#_) << " = " << (_) << endl
#define FAST_IO ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)

const int N=1e5+10;
typedef long long ll;
typedef unsigned long long ull;
int a[N],l[N],r[N];
vector <int> sub[N],add[N];

int main(){
    int mxt=-INF,mit=INF;
    int n,m,ans=-INF,ansid=0,cnt=0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        mit=min(mit,a[i]);mxt=max(mxt,a[i]);
    }
    ans=mxt-mit;
    for(int i=1;i<=m;i++){
        scanf("%d%d",&l[i],&r[i]);
        sub[l[i]].push_back(i);
        add[r[i]+1].push_back(i);
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<sub[i].size();j++){
            int id=sub[i][j];
            for(int k=l[id];k<=r[id];k++) a[k]--;
        }
        for(int j=0;j<add[i].size();j++){
            int id=add[i][j];
            for(int k=l[id];k<=r[id];k++) a[k]++;
        }
        if(sub[i].size()||add[i].size()){
            int mi=INF,mx=-INF;
            for(int j=1;j<=n;j++) mi=min(mi,a[j]),mx=max(mx,a[j]);
            if(ans<mx-mi){
                ans=mx-mi;
                ansid=i;
            }
        }
    }
    printf("%d
",ans);
    for(int i=1;i<=m;i++){
        if(l[i]<=ansid&&r[i]>=ansid) cnt++;
    }
    printf("%d
",cnt);
    for(int i=1;i<=m;i++){
        if(l[i]<=ansid&&r[i]>=ansid) printf("%d ",i);
    }
    return 0;
}