【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;
}