HUAWEI 机试题：数字组合

? Scala 实现

题目描述：

• 给定一个正整数数组，检查数组中是否存在满足规则的数字组合
• 规则： A = B + 2C

输入描述：

• 第一行输出数组的元素个数。
• 接下来一行输出所有数组元素，用空格隔开。

输出描述：

• 如果存在满足要求的数，在同一行里依次输出规则里A/B/C的取值，用空格隔开。
• 如果不存在，输出0。

备注：

1. 数组长度在3-100之间。
2. 数组成员为0-65535，数组成员可以重复，但每个成员只能在结果算式中使用一次。
3. 如：数组成员为[0, 0, 1, 5]，0出现2次是允许的，但结果0 = 0 + 2 * 0是不允许的，因为算式中使用了3个0。
4. 用例保证每组数字里最多只有一组符合要求的解。

示例

输入：

• 4
• 2 7 3 0

输出：

• 7 3 2

说明

• 7 = 3 + 2 * 2

输入：

• 3
• 1 1 1

输出：

• 0

说明：

• 找不到满足条件的组合

代码：

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

    // 获取输入
    val count = StdIn.readLine().toInt
    val input = StdIn.readLine().split(" ").toList.map(_.toInt).sorted
    println(input)
    println(input.length-1)

    // 记录结果
    var res = "0"
    var a, b, c = 0

    // 计算
    for (i <- input.indices.reverse) {
        for (j <- 0 until i) {
            println(input(j))
            for (k <- 0 until i) {
                a = input(i)
                b = input(j)
                c = input(k)
                if (a == b + 2 * c) res = a + " " + b + " " + c
            }
        }
    }

    // 输出结果
    println(res)
}