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

原创 2015年11月21日 19:32:30

本文是《【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位同学最高分。


版权声明:本文为博主原创文章,未经博主允许,欢迎随意转载,标好作者+原文地址就可以了!感谢欣赏!觉得好请回个贴!

相关文章推荐

Java 深拷贝ArrayList实例(包括递归和序列化方法)

前言:       日前一哥们问我一个有关多层ArrayList拷贝的问题,我帮他写了一个例程,感觉以后用得着,便放上来了。如果要在自身类中加入Clone功能,需要implements IClonea...

递归求平方和JAVA实现

  • 2011-05-09 21:49
  • 1010B
  • 下载

java递归解迷宫 求所有解

递归解迷宫加强版本 得到所有的走出迷宫的路 直接代码import java.util.ArrayList; import java.util.List; import java.util.Stack;...

关于海量数据TopK问题的一个具体实现(java)

例题:搜索引擎会通过日志文件把用户每次检索使用的所有检索串都记录下来,每个查询串的长度为1-255字节。假设目前有一千万个记录(这些查询串的重复度比较高,虽然总数是1千万,但如果除去重复后,不超过3百...

Spark下的TopK实现(Java)

import java.io.BufferedReader; import java.io.BufferedWriter; import java.io.FileReader; import java...

Java 实现 堆排序 快速排序 以及 TopK问题(二)

接上文 已知快速排序可以将一个数组分成两部分,一部分大于某个值,一部分小于某个值,那么由这点可以推出取TopK值的方法如下: 假设快排每一趟的分割值的点为p,数组长度为n,那么需要...

Java 利用最小堆解决topK

public class MinHeap { // 堆的存储结构 - 数组 private int[] data; // 将一个数组传入构造方法,并转换成一个小根堆 public MinH...

TopK问题探索-最小堆JAVA实现

TopK问题是指从

TopK问题 Java代码

import java.util.Arrays; public class TopK{ public static void main(String[] args){ int [] data ...

多属性TopK——NRA算法(JAVA)实现

这两天看了一下多属性的TopK,有关NRA,google之。 试着用Java实现了一下,可能有错误,欢迎指出错误。 写的英文注解也不好,凑合着。 算法中使用几种操作:And Or 和加权。 (...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)