🌈 Scala 实现
题目描述:
- 给定一个正整数数组,检查数组中是否存在满足规则的数字组合
- 规则: A = B + 2C
输入描述:
- 第一行输出数组的元素个数。
- 接下来一行输出所有数组元素,用空格隔开。
输出描述:
- 如果存在满足要求的数,在同一行里依次输出规则里A/B/C的取值,用空格隔开。
- 如果不存在,输出0。
备注:
- 数组长度在3-100之间。
- 数组成员为0-65535,数组成员可以重复,但每个成员只能在结果算式中使用一次。
- 如:数组成员为[0, 0, 1, 5],0出现2次是允许的,但结果0 = 0 + 2 * 0是不允许的,因为算式中使用了3个0。
- 用例保证每组数字里最多只有一组符合要求的解。
示例
输入:
- 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)
}
❤️ END ❤️