【Java】求ArrayList中的TopK，不递归

本文是《【Python】求数组（list）中最大的X个数，不递归，尽可能减少代码量》（点击打开链接）的姊妹篇，思想是一模一样的，只是这次是求解一个存放类ArrayList的TopK。

比如有一个存放学生（学号，姓名，分数）实体的ArrayList，里面的学生就乱序的。

我们要找出其中成绩最高的10位学生。

具体请看如下代码，我们先假设这个ArrayList有100个学生，里面的学生的学号分别是1-100，姓名叫stu1-100，分数分别是考1-100分，之后我们用Java自带的洗牌算法打乱其中的排序。用前文的思想同样找其中的TopK。只是在Java中必须用《【Java】Collections中sort方法Comparator的重写》（点击打开链接）中的方法对TopK数组进行排序。

import java.util.*;

class Student {// 学生实体类
	// 类成员
	private int stu_no;
	private String stu_name;
	private double score;

	// getter setter
	public int getStu_no() {
		return stu_no;
	}

	public void setStu_no(int stu_no) {
		this.stu_no = stu_no;
	}

	public String getStu_name() {
		return stu_name;
	}

	public void setStu_name(String stu_name) {
		this.stu_name = stu_name;
	}

	public double getScore() {
		return score;
	}

	public void setScore(double score) {
		this.score = score;
	}
	// ……假设下面还有若干的属性

}

public class TopKTest {

	private static final int K = 10;// 这里的作用就是相当于C语言的#define K 10，const int
									// k=10，反正就是一个常量，你爱怎么理解就怎么理解

	public static void main(String[] args) {
		// 产生一个学生成绩为1-100的数组，并且打乱顺序
		List<Student> stu_list = new ArrayList<Student>();
		for (int i = 1; i <= 100; i++) {
			Student student = new Student();
			student.setStu_no(i);
			student.setStu_name("stu" + i);
			student.setScore(i);
			stu_list.add(student);
		}
		Collections.shuffle(stu_list);

		// 打印给大家看，确实是一个乱序的学生数组
		for (int i = 0; i < stu_list.size(); i++) {
			Student student = stu_list.get(i);
			System.out.print(student.getStu_no() + "," + student.getStu_name()
					+ "," + student.getScore() + " ");
		}
		System.out.println();

		// 之后求最高分那k个学生
		List<Student> top10_stu_list = new ArrayList<Student>();
		for (int i = 0; i < K; i++) {
			top10_stu_list.add(stu_list.get(i));
			Collections.sort(top10_stu_list, new Comparator<Student>() {
				public int compare(Student s1, Student s2) {
					if (s1.getScore() < s2.getScore()) {
						return 1;
					} else if (s1.getScore() > s2.getScore()) {
						return -1;
					} else {
						return 0;
					}
				}
			});
		}

		// 目标数组之后的学生
		for (int i = K; i < stu_list.size(); i++) {
			Student student = stu_list.get(i);
			if (student.getScore() > top10_stu_list.get(K - 1).getScore()) {// 如果此学生大于最大值数组（TopK数组）的最后一个学生的成绩，因为进行过排序，也就是其中的分数最低的学生
				top10_stu_list.remove(K - 1);//那最大值数组（TopK数组）最后一个数，因为进行过排序，也就是其中的最小值出栈
				top10_stu_list.add(student);//此学生进入这个最大值数组（TopK数组）
				Collections.sort(top10_stu_list, new Comparator<Student>() {//再对最大值数组（TopK数组）的学生进行排序
					public int compare(Student s1, Student s2) {
						if (s1.getScore() < s2.getScore()) {
							return 1;
						} else if (s1.getScore() > s2.getScore()) {
							return -1;
						} else {
							return 0;
						}
					}
				});
			}
		}

		//打印最后的结果
		for (int i = 0; i < top10_stu_list.size(); i++) {
			Student student = top10_stu_list.get(i);
			System.out.print(student.getStu_no() + "," + student.getStu_name()
					+ "," + student.getScore() + " ");
		}
		System.out.println();

	}

}

我们的目标是找到其中成绩最高的10位学生。运行结果如下。显然就是分数考91-100分那10位同学最高分。