How to get the median of two sorted slices in Scala

1 Answer

0 votes
object MedianFinder {
  def findMedianOfTwoSortedArrays(arr1: Array[Int], arr2: Array[Int]): Double = {
    val merged = (arr1 ++ arr2).sorted
    val mid = merged.length / 2

    if (merged.length % 2 != 0)
      merged(mid).toDouble
    else
      (merged(mid - 1) + merged(mid)).toDouble / 2
  }

  def main(args: Array[String]): Unit = {
    val arr1 = Array(1, 3, 5)
    val arr2 = Array(2, 4, 6)

    val median = findMedianOfTwoSortedArrays(arr1, arr2)
    println(f"Median: $median%.1f")
  }
}



/*
run:

Median: 3.5

*/

 



answered Sep 23 by avibootz
...