区间交 51Nod - 1672

https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1672

序列元素都是非负数 问题就简单了很多 因为区间长度越长越好 从前往后扫一遍序列 对每个i 看所有覆盖它的线段里左端点第k大是多少 这也就是以i为左端点时 右端点最远能扩展到哪里 线段树离线一下就好

 

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const int maxn=1e5+10;

vector <int> gou[maxn];
int sum[4*maxn];
ll ary[maxn],pre[maxn];
int n,k,m;

void update(int tar,int val,int l,int r,int cur)
{
    int m;
    sum[cur]+=val;
    if(l==r) return;
    m=(l+r)/2;
    if(tar<=m) update(tar,val,l,m,2*cur);
    else update(tar,val,m+1,r,2*cur+1);
}

int query(int tar,int l,int r,int cur)
{
    int m;
    if(l==r) return l;
    m=(l+r)/2;
    if(tar<=sum[2*cur]) return query(tar,l,m,2*cur);
    else return query(tar-sum[2*cur],m+1,r,2*cur+1);
}

int main()
{
    ll ans,res;
    int i,j,l,r;
    scanf("%d%d%d",&n,&k,&m);
    for(i=1;i<=n;i++){
        scanf("%lld",&ary[i]);
        pre[i]=pre[i-1]+ary[i];
    }
    for(i=1;i<=m;i++){
        scanf("%d%d",&l,&r);
        gou[l].pb(l),gou[r+1].pb(-l);
    }
    ans=0;
    for(i=1;i<=n;i++){
        for(j=0;j<gou[i].size();j++){
            if(gou[i][j]>0) update(gou[i][j],1,1,n,1);
            else update(-gou[i][j],-1,1,n,1);
        }
        if(sum[1]>=k){
            res=query(k,1,n,1);
            ans=max(ans,pre[i]-pre[res-1]);
        }
    }
    printf("%lld\n",ans);
    return 0;
}