51nod 1672 区间交【线段树、multiset】

题目：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1672

题意：

小A有一个含有n个非负整数的数列与m个区间，每个区间可以表示为li,ri。
它想选择其中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

一行表示答案

分析：

题解是线段树，也是比较容易的。
看讨论有个很简单的额做法：

可以使用multiset来维护右端点，首先求一下前缀和，之后将区间按照起始时间从小到大排序，对于每个区间，首先将右端点放入multiset，之后删除multiset里比当前的左端点小的值，若multiset里的值多于k个，那么删除到只剩k个，那么当前的最优区间就是[A[i].left,*multiset.begin()]，根据这个值更新ans即可。

代码：

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
typedef pair<int,int>pii;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=100000+7;


ll sum[N];
int q[N];
pii a[N];
int main()
{
    //freopen("in.txt","r",stdin);
    int n,m,k,x;
    scanf("%d%d%d",&n,&k,&m);
    sum[0]=0;
    for(int i=1;i<=n;i++)scanf("%d",&x),sum[i]=sum[i-1]+x;
    for(int i=0;i<m;i++)scanf("%d%d",&a[i].fi,&a[i].se);
    sort(a,a+m);

    ll ans=0;
    multiset<int>mset;
    for(int i=0;i<m;i++){
        mset.insert(a[i].se);
        while(*mset.begin()<a[i].fi)mset.erase(*mset.begin());
        while(mset.size()>k)mset.erase(*mset.begin());
        if(mset.size()==k)ans=max(ans,sum[*mset.begin()]-sum[a[i].fi-1]);
    }
    printf("%lld\n",ans);
    return 0;
}