HUAWEI 机试题：身高体重排序

Scala 实现

题目描述：

• 身高从低到高，身高相同体重从轻到重，体重相同维持原来顺序

示例

输入：

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

输出：

• 2 1 3 4

输入：

• 3
• 90 110 90
• 45 60 45

输出：

• 1 3 2

代码

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

    // 获取输入
    val Num = StdIn.readLine().toInt
    val h_list = StdIn.readLine().split(" ").map(_.toInt).toList
    val w_list = StdIn.readLine().split(" ").map(_.toInt).toList

    // 存放编号及身高体重
    val buffer = new ListBuffer[Tuple3[Int, Int, Int]]
    for (i <- 1 to Num) {
        val tuple = new Tuple3[Int, Int, Int](i, h_list(i - 1), w_list(i - 1))
        buffer.append(tuple)
    }

    // 实现排序
    val resList = buffer.sortWith((o1, o2) => {
        if (!o1._2.equals(o2._2)) {
            o1._2.compareTo(o2._2) < 0
        } else {
            o1._3.compareTo(o2._3) < 0
        }
    })

    // 输出结果
    for (elem <- resList) {
        print(elem._1 + " ")
    }
}

 

---

❤️ END ❤️