小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
#include <iostream>
#include <set>
#include <algorithm>
using namespace std;
typedef long long int ll;
struct node
{
int left, right;
};
const int MAXN = 1e5 + 5;
int input[MAXN];
node nodes[MAXN];
ll sum[MAXN];
int n, k, m;
bool cmp(const node &a, const node &b)
{
return a.left < b.left;
}
int main()
{
cin >> n >> k >> m;
for (int i = 1; i <= n; i++)
{
cin >> input[i];
}
sum[0] = 0;
for (int i = 1; i <= n; i++)
{
sum[i] = sum[i-1] + input[i];
}
for (int i = 1; i <= m; i++)
{
cin >> nodes[i].left >> nodes[i].right;
}
sort(nodes+1, &nodes[m+1], cmp);
multiset<int> buf;
ll result = 0;
for (int i = 1; i <= m; i++)
{
int left = nodes[i].left;
int right = nodes[i].right;
buf.insert(right);
while (*(buf.begin()) < left)
{
buf.erase(buf.begin());
}
if (buf.size() > k)
{
buf.erase(buf.begin());
}
if (buf.size() == k)
{
result = max(result, sum[*(buf.begin())] - sum[left-1]);
}
}
cout << result << endl;
return 0;
}