算法1781：按身高体重排序

题目描述

某学校举行运动会，学生们按编号(1、2、3…n)进行标识，现需要按照身高由低到高排列，对身高相同的人，按体重由轻到重排列；对于身高体重都相同的人，维持原有的编号顺序关系。请输出排列后的学生编号。

输入

两个序列，每个序列由n个正整数组成（0 < n <= 100）。第一个序列中的数值代表身高，第二个序列中的数值代表体重。。

输出

排列结果，每个数值都是原始序列中的学生编号，编号从1开始，身高从低到高，身高相同体重从轻到重，体重相同维持原来顺序。

样例输入

4
100 100 120 130
40 30 60 50

样例输出

2 1 3 4

代码实现

import java.util.*;

public class 身高体重排队 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int[] heights = new int[n];
        int[] weights = new int[n];
        for (int i = 0; i < n; i++) {
            heights[i] = scanner.nextInt();
        }
        for (int i = 0; i < n; i++) {
            weights[i] = scanner.nextInt();
        }
        List<Integer> studentIds = new ArrayList<>();
        for (int i = 1; i <= n; i++) {
            studentIds.add(i);
        }
        Collections.sort(studentIds, new Comparator<Integer>() {
            @Override
            public int compare(Integer id1, Integer id2) {
                int bigHeight = Integer.compare(heights[id1 - 1], heights[id2 - 1]);
                if (bigHeight != 0) {
                    return bigHeight;
                }
                int bigWeight = Integer.compare(weights[id1 - 1], weights[id2 - 1]);
                if (bigWeight != 0) {
                    return bigWeight;
                }
                return Integer.compare(id1, id2);
            }
        });
        for (int id : studentIds) {
            System.out.print(id + " ");
        }
    }
}