Print Diagonal Elements

Given a square matrix the task is to print only the diagonal elements of the matrix.

Example:

571
829
436
Input Matrix

71
8
46
Output Matrix
Java
import java.util.*;
public class PrintDiagonalElements
{
    public static void main(String args[])
    {
        int size=0,temp=0;
        Scanner sc=new Scanner(System.in);
        System.out.print("Enter size of matrix: ");
        size=sc.nextInt();
        int ar[][]=new int[size][size];
        System.out.println("Enter matrix elements:-");
        for(int i=0;i<size;i++)
        {
            for(int j=0;j<size;j++)
            {
                ar[i][j]=sc.nextInt();
            }
        }
        System.out.println("Matrix with diagonal elements:-");
        for(int i=0;i<size;i++)
        {
            for(int j=0;j<size;j++)
            {
                if(i==j||(i+j)==(size-1))
                {
                    System.out.print(ar[i][j]+" ");
                }
                else
                {
                    System.out.print("  ");
                }
            }
            System.out.println();
        }
    }
}
Java