堆排序问题(前500个数)

如何找出排名前500的数

题目:有20个数组,每个数组有500个元素,并且是有序排列好的,现在如何在20*500个数中找到排名前500个数?(假设降序排列)
1.建立大顶堆
2.删除堆顶放入数组
3.重复操作
Comparable接口
此接口强行对实现它的每个类的对象进行整体排序。这种排序被称为类的自然排序,类的 compareTo 方法被称为它的自然比较方法。
int compareTo(T o)
比较此对象与指定对象的顺序。如果该对象小于、等于或大于指定对象,则分别返回负整数、零或正整数。
参数:
o - 要比较的对象。
返回:
负整数、零或正整数,根据此对象是小于、等于还是大于指定对象。
抛出:
ClassCastException - 如果指定对象的类型不允许它与此对象进行比较。
PriorityQueue
一个基于优先级的无界优先级队列。优先级队列的元素按照其自然顺序进行排序,或者根据构造队列时提供的 Comparator 进行排序,具体取决于所使用的构造方法。该队列不允许使用 null 元素也不允许插入不可比较的对象(没有实现Comparable接口的对象)。
PriorityQueue 队列的头指排序规则最小那哥元素。如果多个元素都是最小值则随机选一个。
PriorityQueue 是一个无界队列,但是初始的容量(实际是一个Object[]),随着不断向优先级队列添加元素,其容量会自动扩容,无需指定容量增加策略的细节。
在这里插入图片描述
源码:

import java.util.Arrays;
import java.util.PriorityQueue;

class DataWithSource implements Comparable<DataWithSource>{
	//数据
	private int value;
	//来源的数组
	private int comeFrom;
	//在数组中的index
	private int index;
	
	public DataWithSource(int value,int comeFrom,int index) {
		this.value = value;
		this.comeFrom=comeFrom;
		this.index=index;
	}
	
	public int getValue() {
		return value;
	}

	public void setValue(int value) {
		this.value = value;
	}

	public int getComeFrom() {
		return comeFrom;
	}

	public void setComeFrom(int comeFrom) {
		this.comeFrom = comeFrom;
	}

	public int getIndex() {
		return index;
	}

	public void setIndex(int index) {
		this.index = index;
	}
	
	@Override
	public int compareTo(DataWithSource o) {
		if(this.value>o.getValue())
			return -1;
		else if(this.value<o.getValue())
			return 1;
		else
			return 0;
	}
}

class Test {
	public static int[] getTop(int[][] data) {
		int rowSize = data.length;
		int columnSize = data[0].length;
		int [] result3 = new int[columnSize];
		PriorityQueue<DataWithSource>heap=new PriorityQueue<DataWithSource>();
		for(int i=0;i<rowSize;i++) {
			DataWithSource d=new DataWithSource(data[i][0],i,0);
			heap.add(d);
		}
		int num=0;
		while(num<columnSize) {
			DataWithSource d=heap.poll();
			result3[num++]=d.getValue();
			if(num>=columnSize) {
				break;
			}
			d.setValue(data[d.getComeFrom()][d.getIndex()+1]);
			d.setIndex(d.getIndex()+1);
			heap.add(d);
		}
		return result3;
	}
	public static void main(String[] args) {
		int [][]data= {{29,17,14,2,1},{19,17,16,15,6},{30,25,20,14,5}};
		System.out.println(Arrays.toString(getTop(data)));
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值