How to sort an array that consists of only 0s and 1s in Scala

1 Answer

0 votes
object BinarySorter {

  // Function to sort an array containing only 0s and 1s
  def sortBinaryArray(arr: Array[Int]): Unit = {
    var left = 0                      // Index to track the left side
    var right = arr.length - 1       // Index to track the right side

    while (left < right) {
      // If the left index is at 0, move it forward
      if (arr(left) == 0) {
        println(s"left: $left")
        left += 1
      }
      // If the right index is at 1, move it backward
      else if (arr(right) == 1) {
        println(s"right: $right")
        right -= 1
      }
      // If left is 1 and right is 0, swap them
      else {
        val temp = arr(left)
        arr(left) = arr(right)
        arr(right) = temp
        println(s"swap() left: $left right: $right")
        left += 1
        right -= 1
      }
    }
  }

  def main(args: Array[String]): Unit = {
    // Input: Binary array
    val arr = Array(1, 0, 1, 0, 1, 0, 0, 1, 0)

    // Sort the binary array
    sortBinaryArray(arr)

    // Output the sorted array
    print("Sorted array: ")
    arr.foreach(num => print(s"$num "))
  }
}



/*
run:

swap() left: 0 right: 8
left: 1
right: 7
swap() left: 2 right: 6
left: 3
swap() left: 4 right: 5
Sorted array: 0 0 0 0 0 1 1 1 1

*/

 



answered Sep 3 by avibootz
...