JAVA ProrityQueue的使用：根据身高和姓名排序

一、问题描述

1.1 场景描述：

有一排士兵，需要按照身高进行排序，如果身高相同则按姓名排序。

1.2 输入描述：

第一行一个整数n，表示士兵数

第二行n个空格隔开的正整数h[1, 2, ... n], h[i]表示第i个士兵的高度

第三行n个空格隔开的字符串s[1, 2, ... n]，s[i]表示第i个士兵的姓名。注意，士兵的名字仅包含小写英文字母且没有重复的名字。

n<=50000，h[i]<=300，s[i]仅包含小写英文字母且长度不大于10

1.3 输出描述：

输出一行，包含n个字符串s'[1, 2, ... n]，s'[i]表示经过排序后，从左到右站在第i个位置的士兵的名字，不同名字间用单个空格隔开。

二、思路

很明显，这是一个排序问题，并且排序在前的士兵信息先输出。诶等等，排序在前的士兵先输出...这不就是一个FIFO队列。在FIFO基础上考虑排序的话，这不就是一个PriorityQueue优先队列嘛。

所以，问题的思路很明朗：

1. 创建Scanner对象获取输入，并用n保存人数，height[n]保存身高，name[n]保存姓名。
2. 定义PriorityQueue队列，并重写compareTo函数：若身高相同，根据姓名升序排序，否则依据身高降序排序。
3. 将height，和name存入队列
4. 输出队列

三、代码

import java.util.Objects;
import java.util.PriorityQueue;
import java.util.Scanner;
/**
 * @author Bourne
 * @date 2022-04-09 18:21
 */
public class mt8 {
	/**
	 * @Title: main
	 * @Description:
	 * @param: @param args
	 * @return: void
	 * @throws
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		@SuppressWarnings("resource")
		Scanner scanner = new Scanner(System.in);
		int num = scanner.nextInt();
		int[] height = new int[num];
		String[] name = new String[num];
		for(int i = 0; i < num; i++) height[i] = scanner.nextInt();
		for(int i = 0; i < num; i++) name[i] = scanner.next();
		PriorityQueue<Soldier> priorityQueue = new PriorityQueue<>((e1, e2) ->{
			if(e1.height == e2.height) {
				return e1.name.compareTo(e2.name);
			}
			return e1.height - e2.height;
		});
		for(int i = 0; i < num; i++) priorityQueue.add(new Soldier(height[i], name[i]));
		for(int i = 0; i < num - 1; i++) {
			System.out.print(Objects.requireNonNull(priorityQueue.poll()).name + " ");
		}
		System.out.print(Objects.requireNonNull(priorityQueue.poll()).name);
	}
}
 class Soldier{
	int height;
	String name;
	public Soldier(int height, String name) {
		this.height = height;
		this.name = name;
	}
}