小A有一个含有n个非负整数的数列与m个区间,每个区间可以表示为li,ri。
它想选择其中k个区间, 使得这些区间的交的那些位置所对应的数的和最大。(是指k个区间共同的交,即每个区间都包含这一段,具体可以参照样例)
在样例中,5个位置对应的值分别为1,2,3,4,6,那么选择[2,5]与[4,5]两个区间的区间交为[4,5],它的值的和为10。
Input
第一行三个数n,k,m(1<=n<=100000,1<=k<=m<=100000)。 接下来一行n个数ai,表示小A的数列(0<=ai<=10^9)。 接下来m行,每行两个数li,ri,表示每个区间(1<=li<=ri<=n)。
Output
一行表示答案
Input示例
5 2 3 1 2 3 4 6 4 5 2 5 1 4
Output示例
10
这个题如果会了优先队列的话倒不是很难
先按左端点从小到大排序 然后边把右端点插入最小值优先的优先队列边判断
因为左端点是从小到大的 所以对于当前的每个值左端点是确定的 当目前插入的值小于k时 肯定要继续插入 当等于k时 目前区间数正好 就算一下值 这时的右端点是之前入队中最小的那个。
#include<bits/stdc++.h>
using namespace std;
const int maxn=100005;
struct node
{
int l,r;
}a[maxn];
long long data[maxn],sum[maxn];
int cmp(node a,node b)
{
if(a.l!=b.l)return a.l<b.l;
else return a.r<b.r;
}
int main()
{
int n,k,m;
while(scanf("%d%d%d",&n,&k,&m)!=-1)
{
priority_queue<int ,vector<int>,greater<int> >q;
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
{
scanf("%d",&data[i]);
sum[i]=sum[i-1]+data[i];
}
// cout<<sum[n]<<endl;
for(int i=0;i<m;i++)
scanf("%d%d",&a[i].l,&a[i].r);
sort(a,a+m,cmp);
long long ans=0;
for(int i=0;i<m;)
{
while(q.size()<k)
{
//cout<<"1"<<endl;
q.push(a[i].r);
if(q.size()==k)
ans=max(ans,sum[q.top()]-sum[a[i].l-1]);
//cout<<q.top()<<endl;}
i++;
}
if(q.size()==k)q.pop();
}
printf("%lld\n",ans);
}
return 0;
}