poj 2010 Moo University - Financial Aid (贪心+线段树)

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove 

骗一下访问量。。。。

题意大概是：从c个中选出n个，使得总花费小于等于f，保证价值的中位数最大

http://poj.org/problem?id=2010 

做法：按价值排序之后，枚举中位数，然后对于小于中位数的部分贪心选出花费最小的n/2个，大于的部分也贪心选出花费最小的n/2个，然后比较总花费与f的关系。

可以用两个线段树维护一下，左部分便是每次insert一个，比较当前的第k小的大小，如果更小则remove掉原来的第k小，insert一个新的。

右部分便是每次remove一个，然后更新第k小。

#include <iostream>#include <cmath>#include <cstdio>#include <algorithm>#define lson step<<1#define rson step<<1|1#define LL long longusing namespace std;const int N = 100005;struct Node{int cost,val;bool operator<(const Node n)const{return val<n.val;}}a[N];struct Seg_tree{int left,right;int small,large;}L[N<<2];int m,n,f,x[N],tot;LL Left=0,Right=0;void bulid(int step,int l,int r){L[step].left=l;L[step].right=r;L[step].small=L[step].large=0;if(l==r) return ;int m=(l+r)>>1;bulid(lson,l,m);bulid(rson,m+1,r);}void insert(int step,int pos,int kind){if(!kind) L[step].small++;else L[step].large++;if(L[step].left==pos&&pos==L[step].right) return ;int m=(L[step].left+L[step].right)>>1;if(pos<=m) insert(lson,pos,kind);else insert(rson,pos,kind);}int query(int step,int k,int kind){if(L[step].left==L[step].right) return L[step].left;int m=(L[step].left+L[step].right)>>1;if(!kind){if(L[lson].small>=k) return query(lson,k,kind);else return query(rson,k-L[lson].small,kind);}else{if(L[lson].large>=k) return query(lson,k,kind);else return query(rson,k-L[lson].large,kind);}}void remove(int step,int pos,int kind){if(!kind) L[step].small--;else L[step].large--;if(L[step].left==L[step].right) return ;int m=(L[step].left+L[step].right)>>1;if(pos<=m) remove(lson,pos,kind);else remove(rson,pos,kind);}int main(){scanf("%d%d%d",&m,&n,&f);for(int i=0;i<n;i++){scanf("%d%d",&a[i].val,&a[i].cost);x[i]=a[i].cost;}sort(a,a+n);sort(x,x+n);tot=unique(x,x+n)-x;bulid(1,1,tot);for(int i=0;i<m/2;i++){Left+=(LL)a[i].cost;int pos=lower_bound(x,x+tot,a[i].cost)-x+1;insert(1,pos,0);}for(int i=m/2+1;i<n;i++){int pos=lower_bound(x,x+tot,a[i].cost)-x+1;if(i<m){Right+=(LL)a[i].cost;insert(1,pos,1);}else{int idx=query(1,m/2,1);if(a[i].cost<x[idx-1]) Right=Right-x[idx-1]+a[i].cost;insert(1,pos,1);}}int ans=-1;for(int i=m/2;i<n-m/2;i++){if((LL)Left+Right+a[i].cost<=f){ans=max(ans,a[i].val);}int pos=lower_bound(x,x+tot,a[i].cost)-x+1;int idx=query(1,m/2,0);if(a[i].cost<x[idx-1]){Left-=(LL)x[idx-1];Left+=(LL)a[i].cost;remove(1,idx,0);insert(1,pos,0);}pos=lower_bound(x,x+tot,a[i+1].cost)-x+1;remove(1,pos,1);idx=query(1,m/2,1);if(a[i+1].cost<x[idx-1]){Right-=(LL)a[i+1].cost;Right+=(LL)x[idx-1];}}printf("%d\n",ans);    return 0;}