简单的数据结构题
由于所有的数都是非负整数,所以对于一个确定的左端点,区间交最大时右端点尽可能向右取;
先将所有的区间按照 左端点 从小到大 排序;
枚举区间交的左端点,那么该区间交最大时的右端点 是 所有区间中 左端点在枚举端点的左侧的那些区间中 从右往左数第k个右端点;
因为是从左往右枚举,所以符合条件的区间会越来越多,已经符合条件的区间会一直符合条件;
维护一个 大小为k 的大根堆,记录已经符合条件区间中最靠近右边的k个区间右端点
每次加入一个新的右端点,和堆顶的元素比较,如果新右端点更优则弹出堆顶加入新元素,并且重新平衡堆。
用sum[堆顶元素]减去sum[枚举]更新答案。
O( N * log N)
#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#define N 100050
using namespace std;
struct A {int l,r;}a[N];
bool cmp(A p1,A p2) {if (p1.l == p2.l) return p1.r < p2.r; else return p1.l < p2.l;}
priority_queue<int,vector<int>,greater<int> > q;
int n,m,k,v[N];
long long sum[N],ans = 0LL;
int main() {
scanf("%d%d%d",&n,&k,&m);
for (int i=1;i<=n;i++) scanf("%d",&v[i]);
for (int i=1;i<=n;i++) sum[i] = sum[i-1] + 1LL*v[i]; //前缀和
for (int i=1;i<=m;i++) scanf("%d%d",&a[i].l,&a[i].r);
sort(a+1,a+m+1,cmp);//左端点为第一关键词,右端点为第二关键词 排序
for (int i=1;i<=k;i++) q.push(a[i].r);//将前k个区间右端点加入优先队列
int t = k+1;
for (int i=a[k].l;i<=n;i++) {//小于a[k].l的点是无法作为左端点的
while (a[t].l == i) {
int u = q.top();
if (u < a[t].r) q.pop() , q.push(a[t].r);
t++;
}
int u = q.top();
ans = max(ans,sum[u] - sum[i-1]);
}
printf("%I64d\n",ans);
return 0;
}