华为OD机试真题-按身高和体重排序

华为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来对学生列表进行排序。排序规则是：首先按照身高升序排列，如果身高相同，则按照体重升序排列，如果身高和体重都相同，则保持原有的编号顺序关系。最后，程序遍历排序后的学生列表，按顺序输出学生的编号。