华为OD机试真题-按身高和体重排序
题目描述:
某学校举行运动会,学生们按编号(1、2、3...n)进行标识,现需要按照身高由低到高排列,对身高相同的人,按体重由轻到重排列;对于身高体重都相同的人,维持原有的编号顺序关系。请输出排列后的学生编号。
输入描述:
两个序列,每个序列由n个正整数组成(0<n<=100)。第一个序列中的数值代表身高,第二个序列中的数值代表体重。
输出描述:
排列结果,每个数值都是原始序列中的学生编号,编号从1开始
示例1
输入
4100 100 120 13040 30 60 50
输出
2 1 3 4
说明
输出的第一个数字2表示此人原始编号为2,即身高为100,体重为30的这个人。由于他和编号为1的人身高一样,但体重更轻,因此要排在1前面。
Java代码实现
package org.example.od;import java.util.*;public class To3 {private static class Student{private int height;private int weight;private int id;public Student(int height, int weight, int id) {this.height = height;this.weight = weight;this.id = id;}}public static void main(String[] args) {Scanner scanner = new Scanner(System.in);List<Student> students = new ArrayList<>();int n = scanner.nextInt();int[] height = new int[n];int[] weight = new int[n];for (int i = 0; i < n; i++) {height[i] = scanner.nextInt();}for (int i = 0; i < n; i++) {weight[i] = scanner.nextInt();}for (int i = 0; i < n; i++) {students.add(new Student(height[i], weight[i], i+1));}// 根据身高升序,身高相同则按体重升序,再相同则保持原编号顺序students.sort((o1, o2) -> {if (o1.height != o2.height) {return o1.height - o2.height;} else if (o1.weight != o2.weight) {return o1.weight - o2.weight;} else {return o1.id - o2.id;}});for (Student student : students){System.out.println(student.id);}}
}
解题思路:
先读取学生数量,然后分别读取学生的身高和体重信息。接着,它创建了一个
Student
对象列表,并将每个学生的相关信息存储在这些对象中。之后,使用Collections.sort
方法和一个自定义的Comparator
来对学生列表进行排序。排序规则是:首先按照身高升序排列,如果身高相同,则按照体重升序排列,如果身高和体重都相同,则保持原有的编号顺序关系。最后,程序遍历排序后的学生列表,按顺序输出学生的编号。