关闭

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

标签: JavaTopK算法最值ArrayList
576人阅读 评论(0) 收藏 举报
分类:

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


0
0

猜你在找
【直播】计算机视觉原理及实战—屈教授
【套餐】深度学习入门视频课程—唐宇迪
【套餐】Hadoop生态系统零基础入门--侯勇蛟
【套餐】嵌入式Linux C编程基础--朱有鹏
【套餐】2017软考系统集成项目——任铄
【套餐】Android 5.x顶级视频课程——李宁
【直播】广义线性模型及其应用——李科
【直播】从0到1 区块链的概念到实践
【直播】机器学习之凸优化——马博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:1205208次
    • 积分:19330
    • 等级:
    • 排名:第438名
    • 原创:716篇
    • 转载:0篇
    • 译文:0篇
    • 评论:287条
    文章分类
    【备注】博文GIF动画的录制方法
    先用屏幕录制软件Freez Screen Video Capture录制屏幕,再把得到的AVI,利用GIF Movie Gear转化成GIF
    【友情链接】亲笔小说