HUAWEI 机试题：最大最小N个数的和

🌈 Scala 实现

题目描述：

• 给定一个数组，编写一个函数来计算它的最大N个数与最小N个数的和。你需要对数组进行去重。

题目说明：

• 数组中数字范围[0, 1000]
• 最大N个数与最小N个数不能有重叠，如有重叠，输入非法返回-1
• 输入非法返回-1

输入描述：

• 第一行输入M， M标识数组大小
• 第二行输入M个数，标识数组内容
• 第三行输入N，N表达需要计算的最大、最小N个数

输出描述：

输出最大N个数与最小N个数的和。

示例

输入：

5
95 88 83 64 100
2

输出：

342

说明：最大2个数[100,95],最小2个数[83,64], 输出为342

输入：

5
3 2 3 4 2 2

输出：

-1

说明：最大2个数[4,3],最小2个数[3,2], 有重叠输出为-1

代码：

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

    // 获取输入
    val M = StdIn.readLine().toInt
    val Mints = StdIn.readLine().split(" ").map(_.toInt).toSet.toList
    val N = StdIn.readLine().toInt

    // 记录结果
    var res = -1

    println(Mints)

    // 计算最大N个数和最小N个数的和
    if (Mints.size >= 2 * N) {
        res = 0
        for (i <- Mints.indices) {
            if (i < N | i > Mints.size - 1 - N) {
                res += Mints(i)
            }
        }
    }

    // 输出结果
    println(res)
}

 

---

❤️ END ❤️