Sum of left & Right diagonal of the matrix In Swift

Sum of left & Right diagonal of the matrix In Swift

  import Foundation

  class SumDiagonal{

    

    // Left Sum Diagonal

    func sumOfLeftDiagonal(mxt: [[Int]], size: Int) -> Int{

        var sum = 0

        for x in 0..<size{

            for y in 0..<size{

                if x == y{

                    sum += mxt[x][y]

                }

            }

        }

        return sum

    }

    

    //MARK: - Right Sum Diagonal

    func sumOfRightDiagonal(mxt: [[Int]], size: Int) -> Int{

        var sum = 0

        for x in 0..<size{

            for y in 0..<size{

                if (x+y) == size - 1{

                    sum += mxt[x][y]

                }

            }

        }

        return sum

    }

 }

 let matrix = [[3, 5, 9, 1], [3, 5, 7, 3], [3, 2, 6, 1], [4, 5, 2, 6]]

 //MARK: - Print Matrix Format 4X4

 for x in 0..<4{

    for y in 0..<4{

        print(matrix[x][y], terminator: " ")

    }

    print()

 }

 //MARK: - Output === >

 /*

 3 5 9 1

 3 5 7 3

 3 2 6 1

 4 5 2 6

 */

 //MARK: - Class object

 let objSumDiagonal = SumDiagonal()

 //MARK: - Now Sum of Left Diagonal

 print("Sum of Left Diagonal == >>    \(objSumDiagonal.sumOfLeftDiagonal(mxt: matrix,   size:  4))")

 //MARK: - Output === >

 //Sum of Left Diagonal == >>    20

 //MARK: - Now Sum of Right Diagonal

 print("Sum of Right Diagonal == >>    \(objSumDiagonal.sumOfRightDiagonal(mxt: matrix,   size: 4))")

 //MARK: - Output === >

 //Sum of Right Diagonal == >>    14