hdu 5700 区间交 (线段树＋枚举)

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=5700

题意：

给定序列a1,a2...an以及m个区间，从m个区间中选出k个区间使k个区间相交的部分和最大。

思路：

预处理前缀和。

按照r从小到大l从大到小的顺序将所有区间排序，用线段树维护区间第k小。然后枚举区间的右端点，求出当前第k小左端点，然后计算差值即可。

#include <iostream>
#include <cstdio>
#include <string.h>
#include <algorithm>
#include <cmath>
#include <map>
#include <climits>
using namespace std;

#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r

typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn=1e5+10;
int T,n,m,k;
int a[maxn];
ll sum[maxn];
int v[maxn<<2];

struct Interval{
	int l;
	int r;
	bool operator < (const Interval &a) const {
		return r<a.r||(r==a.r&&l>a.l);
	}
}p[maxn];

void PushUp(int rt){
	v[rt]=v[rt<<1]+v[rt<<1|1];
}

void Update(int rt,int l,int r,int pos){
	if(l==r){
		v[rt]++;
		return ;
	}
	int mid=(l+r)>>1;
	if(pos<=mid) Update(lson,pos);
	else Update(rson,pos);
	PushUp(rt);
}

int Query(int rt,int l,int r,int k){
	if(l==r) return l;
	int mid=(l+r)>>1;
	if(k<=v[rt<<1]) return Query(lson,k);
	else return Query(rson,k-v[rt<<1]);
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
    while(~scanf("%d%d%d",&n,&k,&m)){
    	memset(v,0,sizeof(v));
    	memset(sum,0,sizeof(sum));
    	for(int i=1;i<=n;i++) {
    		scanf("%d",&a[i]);
    		sum[i]=sum[i-1]+a[i];
    	}
    	for(int i=1;i<=m;i++) scanf("%d%d",&p[i].l,&p[i].r);
    	sort(p+1,p+m+1);
    	ll ans=0;
    	for(int i=m;i>=1;i--){
    		Update(1,1,n,p[i].l);
    		if(i<=m-k+1){
    			int l=Query(1,1,n,k);
    			if(l<=p[i].r)
    				ans=max(ans,sum[p[i].r]-sum[l-1]);
    		}
    	}
    	printf("%lld\n",ans);
    }
    return 0;
}