小A有一个含有n个非负整数的数列与m个区间,每个区间可以表示为li,ri。
它想选择其中k个区间, 使得这些区间的交的那些位置所对应的数的和最大。(是指k个区间共同的交,即每个区间都包含这一段,具体可以参照样例)
在样例中,5个位置对应的值分别为1,2,3,4,6,那么选择[2,5]与[4,5]两个区间的区间交为[4,5],它的值的和为10。
输入
第一行三个数n,k,m(1<=n<=100000,1<=k<=m<=100000)。
接下来一行n个数ai,表示小A的数列(0<=ai<=10^9)。
接下来m行,每行两个数li,ri,表示每个区间(1<=li<=ri<=n)。
输出
一行表示答案
输入样例
5 2 3
1 2 3 4 6
4 5
2 5
1 4
输出样例
10
将给的区间根据左端点非递减排序,用排序后的前k个区间的右端点建树
分别将从第k个区间到第m个区间的左端点作为区间交的左端点
从树里寻找满足条件的最优右端点
每次再将下一个点的右端点放入树里
详见代码
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <vector>
typedef long long ll;
using namespace std;
const int maxn=1e5+10;
struct node
{
int sta,en;
}a[maxn];
struct tree
{
int l,r;
int cover;
}t[maxn<<2];
bool com(node a,node b)
{
return a.sta<b.sta;
}
void build(int l,int r,int p)
{
t[p].l=l;t[p].r=r;
if(l==r)
{
t[p].cover=0;
return ;
}
int mid=(l+r)>>1;
build(l,mid,p<<1);
build(mid+1,r,p<<1|1);
t[p].cover=t[p<<1].cover+t[p<<1|1].cover;
}
void update(int p,int x)
{
if(t[p].l==t[p].r)
{
t[p].cover++;
return ;
}
int mid=(t[p].l+t[p].r)>>1;
if(x<=mid)
update(p<<1,x);
else
update(p<<1|1,x);
t[p].cover=t[p<<1].cover+t[p<<1|1].cover;
}
int query(int p,int ask)
{
if(t[p].l==t[p].r)
return t[p].l;
if(t[p<<1|1].cover>=ask)
query(p<<1|1,ask);
else
query(p<<1,ask-t[p<<1|1].cover);
}
int n,k,m;
ll sum[maxn],num,ans;
int main()
{
scanf("%d%d%d",&n,&k,&m);
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
{
scanf("%lld",&num);
sum[i]+=sum[i-1]+num;
}
for(int i=1;i<=m;i++)
scanf("%d%d",&a[i].sta,&a[i].en);
sort(a+1,a+m+1,com);
build(1,n,1);//给区间右边界建树
for(int i=1;i<=k;i++)
update(1,a[i].en);//更新前k个数的右端点
ans=0;
for(int i=k;i<=m;i++)
{
int pos=query(1,k);//询问最优右端点
ans=max(ans,sum[pos]-sum[a[i].sta-1]);
update(1,a[i+1].en);//更新下一个点的右端点
}
printf("%lld",ans);
return 0;
}