华为OD机试真题-按身高和体重排序
题目描述:
某学校举行运动会,学生们按编号(1、2、3...n)进行标识,现需要按照身高由低到高排列,对身高相同的人,按体重由轻到重排列;对于身高体重都相同的人,维持原有的编号顺序关系。请输出排列后的学生编号。
输入描述:
两个序列,每个序列由n个正整数组成(0<n<=100)。第一个序列中的数值代表身高,第二个序列中的数值代表体重。
输出描述:
排列结果,每个数值都是原始序列中的学生编号,编号从1开始
示例1
输入
4 100 100 120 130 40 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
来对学生列表进行排序。排序规则是:首先按照身高升序排列,如果身高相同,则按照体重升序排列,如果身高和体重都相同,则保持原有的编号顺序关系。最后,程序遍历排序后的学生列表,按顺序输出学生的编号。