HUAWEI 机试题：按身高和体重排队

🌈 Scala 实现

题目描述：

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

输入描述：

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

输出描述：

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

示例：

输入：

• 4
• 100 100 120 130
• 40 30 60 50

输出：

• 2134

输入：

• 3
• 90 110 90
• 45 60 45

输出：

• 132

代码：

def main(args: Array[String]): Unit = {

    val scan = new Scanner(System.in)
    // 获取总人数
    val count = scan.nextLine().toInt
    // 获取每个人的身高
    val height = scan.nextLine().split(" ").map(_.toInt)
    // 获取每个人的体重
    val weight = scan.nextLine().split(" ").map(_.toInt)

    // 存放学生对象
    val stuList = ListBuffer[Tuple3[Integer, Integer, Integer]]()
    // 添加学生对象
    for (i <- 1 to count) {
        stuList.append(Tuple3(i, height(i - 1), weight(i - 1)))
    }

    // 实现排序
    stuList
    .sortWith((o1, o2) => {
        if (!o1._2.equals(o2._2)) {
            o1._2.compareTo(o2._2) < 0 // 按照身高由低到高排列
        } else if (!o1._3.equals(o2._3)) {
            o1._3.compareTo(o2._3) < 0 // 对身高相同的人，按体重由轻到重排列
        } else {
            o1._1.compareTo(o2._1) < 0 // 对于身高体重都相同的人，维持原有的编号顺序关系
        }
    })
    .foreach(id => print(id._1)) // 输出排序后的结果
}

 

---

❤️ END ❤️