Question

How to check whether a matrix is a magic square or not in Scala

Answer 1

object MagicSquareChecker {

  // Function to check if the matrix is a magic square
  def isMagicSquare(matrix: Array[Array[Int]]): Boolean = {
    val size = matrix.length
    var sumDiagonal1 = 0
    var sumDiagonal2 = 0

    // Calculate the sum of the primary diagonal
    for (i <- 0 until size) {
      sumDiagonal1 += matrix(i)(i)
    }

    // Calculate the sum of the secondary diagonal
    for (i <- 0 until size) {
      sumDiagonal2 += matrix(i)(size - i - 1)
    }

    // If the two diagonals don't have the same sum, it's not a magic square
    if (sumDiagonal1 != sumDiagonal2) {
      return false
    }

    // Check sums of each row and column
    for (i <- 0 until size) {
      var sumRow = 0
      var sumCol = 0

      for (j <- 0 until size) {
        sumRow += matrix(i)(j)  // Sum of the current row
        sumCol += matrix(j)(i)  // Sum of the current column
      }

      // If any row or column sum is not equal to the diagonal sum, it's not a magic square
      if (sumRow != sumDiagonal1 || sumCol != sumDiagonal1) {
        return false
      }
    }

    // If all checks pass, it's a magic square
    true
  }

  def main(args: Array[String]): Unit = {
    val matrix = Array(
      Array(8, 3, 4),
      Array(1, 5, 9),
      Array(6, 7, 2)
    )

    if (isMagicSquare(matrix)) {
      println("The given matrix is a magic square.")
    } else {
      println("The given matrix is NOT a magic square.")
    }
  }
}



/*
run:

The given matrix is a magic square.

*/

 

70+ SQL courses for beginners and professionals