Two Sum (Scala)

最新推荐文章于 2025-12-15 22:13:28 发布

原创最新推荐文章于 2025-12-15 22:13:28 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#scala #leetcode

Leetcode Scala 专栏收录该内容

1 篇文章

订阅专栏

本文解析了如何使用Scala编写的高效算法解决LeetCode问题'两数之和'，通过zipWithIndex操作、Map数据结构以及collectFirst方法，展示了如何在O(n)复杂度下找到数组中和为目标值的两个数。

def twoSum(nums: Array[Int], target: Int): Array[Int] = {
    val numsIndex = nums.zipWithIndex
    val map = numsIndex.toMap
    numsIndex
        .collectFirst { 
            case (num, index) if(map.get(target - num).exists(_ != index)) => 
                Array(index, map(target - num))
        }.getOrElse(Array(-1, -1))
}

Original code enhancement reference:

https://leetcode.com/problems/two-sum/discuss/260824/Scala-Solution