大顶堆应用:POJ2010

POJ2010题意:
    奶牛学校招生,c头奶牛报名,要选n头(n为奇数),学校是义务制,所以每头奶牛的学费都由学校负责。每头奶牛都由自己的考试分数和它需要花的学费,学校总共有f的资金,问合法招生方案中中间分数(即排名第(n+1)/2)最高的是多少。

题解:先将所有的奶牛按照分数由低到高排序,假设k是招的奶牛中排名中间的那头,按照排序可知,[1,k-1]中的奶牛必定被招了(n-1)/2头,[k+1,c]中也必定被招了(n-1)/2头,而且无论招的是谁,分数是怎么样,最后影响结果的都只是k的分数。于是,可以预处理low[i]代表[1,i]头牛中选出(n-1)/2头牛的最小花费,high[i]代表[i,c]头牛中选出(n-1)/2头牛的花费,预处理方法可以用一个大顶堆,复杂度nlogn,最后枚举中间牛复杂度n。


【输入】

第一行n、c、f

接下来c行每行两个数字,分别为分数和学费

【输出】

合法招生方案中中位数分数(即排名第(n+1)/2)最高的是多少。

样例:
Sample Input

3 5 70
30 25
50 21
20 20
5 18
35 30

Sample Output

35


AC源码:
import java.util.*;
class node implements Comparable { 
    int score, need; 

    public int compareTo(Object o) {//按score升序排列  
       node b=(node)o;    
       return this.score-b.score;   
                    
   }         

 }

  public class Main{ 
   node[] p; //所有牛的数组
   int n, c, f;//题目中给出的三个条件
   int[] h;//维护学费的堆
   int r;//堆底的索引
   int sum; //记录堆内元素之和
   int[] high, low; 
  

 
  void up(int i){//向上调整 
    int j; 
    while(i > 1){ 
        j = i / 2;//i的父亲 
        if(h[i] > h[j]) swap(i, j); 
        else break; 
        i = j; 
    } 
  }
 
  void down(int i){ //向下调整
    int j; 
    while(i * 2 <= r){ 
      j = i * 2; //i的左儿子
      if(j + 1 <= r && h[j + 1] > h[j]) j++; 
        if(h[j] > h[i]) swap(i, j); 
        else break; 
        i = j; 
    } 
} 
 
  void del(int x){ //删除堆顶,将x作为堆顶
    sum = sum + x - h[1]; 
    h[1] = x; 
    down(1); 
  }
  
  void insert(int x){//在堆底插入 
    h[++r] = x; 
    sum += x; 
    up(r); 
  }

  
    //交换
    private void swap(int i, int j) {
       int temp = h[i];
       h[i] = h[j];
       h[j] = temp;
    }

    public void print(){
      for(int i=1;i<=c;i++){
         System.out.print("["+p[i].score+","+p[i].need+"]  ");
     }
    }

    public static void main(String args[]){
         Main ma=new Main();
         ma.go();
    }

   public void go(){
      Scanner in=new Scanner(System.in); 
       n =in.nextInt(); //n是奇数
       c=in.nextInt();
       f=in.nextInt();
       low=new int[c+1];
       high=new int[c+1];
       h=new int[c+1];
       p=new node[c+1];
     for(int i = 1; i <= c; i++){
       p[i]=new node();
       p[i].score=in.nextInt();
       p[i].need=in.nextInt();
     }
    r = sum = 0; 
    Arrays.sort(p, 1, c+1); //按分数升序
    n /= 2; 
    for(int i = 1; i <= n; i++) insert(p[i].need); //从开头往后算n/2个牛的学费
    low[n] = sum; //记录前面n/2个学费之和

    for(int i = n + 1; i <= c - n; i++) 
    { 
        if(p[i].need < h[1]) del(p[i].need); //p[i]的学费比堆顶小,进堆并删除原堆顶
        low[i] = sum; //记录前i个牛中取n/2个时的最小学费和
    } 
    h=new int[c+1]; 
    r = sum = 0; 
    for(int i = c; i > c - n; i--) insert(p[i].need); //从后面往前算n/2个牛
    high[c - n + 1] = sum; //记录最小学费和
    for(int i = c - n; i > n; i--) 
    { 
        if(p[i].need < h[1]) del(p[i].need); //p[i]的学费比堆顶小,进堆并删除原堆顶
        high[i] = sum; //记录从i开始后面的所有牛中取n/2个时的最小学费和
    } 
    int ans = -1; 
    for(int i = c - n; i > n; i--) //枚举中位数,选分数最大的,注意:已经按分数升序排了.
        if(low[i - 1] + high[i + 1] + p[i].need <= f) 
        { 
            ans = p[i].score; 
            break; 
        } 
    System.out.println(ans); 
   
   } 
}


你可能感兴趣的:(java,数据结构,编程,算法)