Scala代码数值比较异常：Advent of Code 2021 Day1与Python结果不符

问题根源 #

你遇到的数值比较异常，核心原因是Scala代码直接对字符串做了比较，而非数值比较：

• Source.fromFile(...).getLines()读取的每行内容都是String类型，字符串比较是按字典序（逐字符对比ASCII值）进行的，不是数值大小。
• 比如"1001"和"992"对比时，第一个字符'1'的ASCII值小于'9'，所以会判定"1001" > "992"为false；同理"996" > "1007"会因为首字符'9' > '1'返回true，这和数值大小的逻辑完全不符。
• 而Python代码中，pandas.read_csv会自动将列解析为数值类型（如int），所以是正确的数值比较逻辑。

修复后的Scala代码 #

只需把读取到的字符串转换为整数类型再进行比较即可：

import scala.io.Source

object advent_of_code_2021_01 {
    def main(args: Array[String]): Unit = {
        // 将每行字符串转为Int类型，确保数值比较逻辑正确
        val numbers = Source.fromFile("01_input.csv").getLines().map(_.toInt).toList
        var increases = 0

        for (i <- 1 until numbers.length) {
            if (numbers(i) > numbers(i-1)) {
                increases += 1
                println(s"$i-${numbers(i)}-${numbers(i-1)}")
            }
        }

        println(increases)
    }
}

额外优化（Scala函数式风格） #

Scala更推荐用函数式写法替代循环，比如用sliding方法简化相邻元素对比逻辑：

import scala.io.Source

object advent_of_code_2021_01 {
    def main(args: Array[String]): Unit = {
        val numbers = Source.fromFile("01_input.csv").getLines().map(_.toInt).toList
        // 滑动窗口取相邻元素对，直接统计后一个大于前一个的数量
        val increases = numbers.sliding(2).count { case List(prev, curr) => curr > prev }
        println(increases)
    }
}

内容的提问来源于stack exchange，提问作者Zachary Brasseaux