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 long
using 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;
}
上一篇:BC 2015在百度之星程序设计大赛 - 预赛(1)(KPI-树董事长)


下一篇:利用python代码获取文件特定的内容,并保存为文档