【51nod1672】区间交

题目大意：给定一个长度为 N 的序列，以及 M 个区间，现从中选出 K 个区间，使得这些区间的交集区间的点权和最大，求最大值是多少。

题解：
发现直接选 K 个区间不可做，考虑从答案入手。设答案区间为 [l,r]，进行枚举答案区间的左端点。当前枚举到的左端点设为 L，那么能够以 L 作为左端点的区间一定满足左端点不超过 L，且右端点大于等于 L。考虑若有超过 K 个区间符合要求，那么肯定是选取较大的 K 个区间的答案更优，因此只需求出符合条件的区间右端点的第 K 大值，并更新答案即可。再考虑 L 之间的转移带来的变化，枚举到 L 时，应该将符合要求的答案更新；同样，统计完 L 的贡献之后，应该将对于 L+1 不符合情况的解删去。需要维护一个支持插入删除和求第 K 大的数据结构，显然权值线段树符合要求。

代码如下

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

int n,m,K,sz[maxn<<2];
LL a[maxn],sum[maxn],ans;
vector<int> st[maxn],ed[maxn];
void insert(int o,int l,int r,int pos,int val){
    if(l==r){sz[o]+=val;return;}
    int mid=l+r>>1;
    if(pos<=mid)insert(o<<1,l,mid,pos,val);
    else insert(o<<1|1,mid+1,r,pos,val);
    sz[o]=sz[o<<1]+sz[o<<1|1];
}
int kth(int o,int l,int r,int k){
    if(l==r)return l;
    int mid=l+r>>1;
    if(sz[o<<1|1]>=k)return kth(o<<1|1,mid+1,r,k);
    else return kth(o<<1,l,mid,k-sz[o<<1|1]);
}

void read_and_parse(){
    scanf("%d%d%d",&n,&K,&m);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    for(int i=1;i<=m;i++){
        int l,r;
        scanf("%d%d",&l,&r);
        st[l].pb(r),ed[r].pb(r);
    }
}
void solve(){
    for(int i=1;i<=n;i++){
        for(auto r:st[i])insert(1,1,n,r,1);
        if(sz[1]>=K){
            int pos=kth(1,1,n,K);
            ans=max(ans,sum[pos]-sum[i-1]);
        }
        for(auto r:ed[i])insert(1,1,n,r,-1);
    }
    printf("%lld\n",ans);
}
int main(){
    read_and_parse();
    solve();
    return 0;
} 

原文地址：https://www.cnblogs.com/wzj-xhjbk/p/11074078.html