Golang Program to Compute the Sum of Diagonals of a Matrix

In this tutorial, we will write a go language program to find the sum of diagonal elements of a matrix.

Compute the Sum of Diagonals of a Matrix

In this example, we will find the sum of left diagonal elements of a 3 x 3 matrix using external function.

Algorithm

Step 1 ? Import the fmt package.

Step 2 ? Create a function to find the sum of the given matrix.

Step 3 ? In this function initialize an integer variable named sum and use a for loop to intend over the matrix array.

Step 4 ? In each iteration update the sum variable with the diagonal elements of matrix (like 00, 11, 22) and return the sum variable.

Step 5 ? Start the main function. Here, first we need to initialize a matrix and print it on the screen using for loops.

Step 6 ? Now, call the matrixSum() by passing the matrix as argument.

Step 7 ? Store the result in result variable and print it on the screen using fmt.Println() function.

Example

package main import "fmt" // function to find sum func matrixSum(mat [][]int, n int) int { var sum int = 0 for k := range mat { sum = sum + mat[k][k] } return sum } func main() { mat := [][]int{ {10, 1, 2}, {4, 5, 6}, {8, 9, 10}, } fmt.Println("The given matrix is: \n") for i := 0; i < 3; i++ { for j := 0; j < 3; j++ { fmt.Print(mat[i][j], "\t") } fmt.Println() } fmt.Println() var n int = 2 result := matrixSum(mat, n) fmt.Println("The Sum of left diagonal element of Matrix = ", result) } 

Output

The given matrix is: 10	1	2 4	5	6 8	9	10 The Sum of left diagonal element of Matrix = 25 

Find the Sum of the Right Diagonal of a 3 x 3 Matrix

In this example, we will write a go language program to find the sum of the right diagonal element of a 3 X 3 matrix.

Algorithm

Step 1 ? Import the fmt package.

Step 2 ? Create a function to find the sum of the given matrix.

Step 3 ? In this function initialize an integer variable named sum and used a for loops to intend over the matrix array.

Step 4 ? In each iteration update the sum variable with the diagonal elements of matrix (like 02, 11, 20) and return the sum variable.

Step 5 ? To obtain this we have used a if condition which satisfies the above condition by computing the sum of iterated variables to 2.

Step 6 ? Start the main function. Here, first we need to initialize a matrix and print it on the screen using for loops.

Step 7 ? Now, call the matrixSum() by passing the matrix as argument.

Step 8 ? Store the result in result variable and print it on the screen using fmt.Println() function.

Example

package main import "fmt" // function to find sum func findSum(mat [][]int, n int) int { var sum int = 0 for i := 0; i < 3; i++ { for j := 0; j < 3; j++ { if i+j == 2 { sum = sum + mat[i][j] } } } return sum } // calling main() func main() { mat := [][]int{ {1, 1, 2}, {4, 5, 6}, {8, 9, 3}, } fmt.Println("The given matrix is: \n") for i := 0; i < 3; i++ { for j := 0; j < 3; j++ { fmt.Print(mat[i][j], "\t") } fmt.Println() } fmt.Println() var n int = 3 result := findSum(mat, n) fmt.Printf("\nSum of right diagonal elements is: %d", result) } 

Output

The given matrix is: 1	1	2 4	5	6 8	9	3 Sum of right diagonal elements is: 15 

Conclusion

We have successfully compiled and executed a go language program to find the sum of diagonal elements of a 3 X 3 matrix along with examples. In the first example, we found the sum of left diagonal of a matrix and in the second example, we found the sum of right diagonal of matrix respectively.