区间交

来源:互联网 发布:2017勇士vs骑士g3数据 编辑:程序博客网 时间:2024/04/25 21:33
小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 31 2 3 4 64 52 51 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;}