用PriorityQueue解决选择最小的K个数问题

Java没有堆的实现，现场写也来不及，有的文献说用TreeSet，比如剑指offer，但是TreeSet是一个set，相同的数只能存一个，个人感觉不合适，相比之下，Java中的PriorityQueue倒是一个不错的选择。用PriorityQueue的实现的代码如下：

/**
	 * 用PriorityQueue实现选择最小的k个数
	 * @param array	数组
	 * @param k
	 * @return
	 */
	public int[] selectKmin(int[] array,int k){
		int[] res = new int[k];	
		//创建一个降序排列的PriorityQueue，自定义比较器作为参数
		PriorityQueue<Integer> priorityQueue = new PriorityQueue<Integer>(k,new Comparator<Integer>(){
 
			@Override
			public int compare(Integer o1, Integer o2) {
				return o2-o1;
			}});
		
		for(int i=0;i<array.length;i++){
			if(priorityQueue.size()<k){
				priorityQueue.add(array[i]);
			}else{
				int maxInQueue = priorityQueue.peek();
				if(maxInQueue>array[i]){//每次取数与堆顶的元素进行比较，
				//如果堆顶元素大，则删除堆顶元素，并添加这个新数到堆中
					priorityQueue.poll();
					priorityQueue.add(array[i]);
				}
			}
		}
		
		//升序存放
		for(int i=0;i<k;i++){
			res[k-i-1] = priorityQueue.poll();
		}
		
		return res;
	}