Заполнить матрицу линейной последовательностью чисел, от левого верхнего угла по диагонали: вправо - вверх

import java.util.Scanner;
 
public class DiagonalFiller {
    public static void fillDiagonally(int[][] matrix) {
        int size = matrix.length;
        int current = 1;
        for (int i = 0; i < size; i++) 
            for (int j = 0, k = i; j <= i; j++, k--)
                matrix[k][j] = current++;
        for (int j = 1; j < size; j++)
            for (int i = size-1, k = j; i >= j; i--, k++)
                matrix[i][k] = current++;
    }
    
    public static void showMatrix(int[][] matrix) {
        int size = matrix.length;
        int maxNumber = size*size;
        int maxNumberLength = 0;
        while (maxNumber > 0) {
            maxNumber /= 10;
            maxNumberLength++;
        }
        for (int i = 0; i < size; i++) {
            for (int j = 0; j < size; j++)
                System.out.printf("%" + maxNumberLength + "d ", matrix[i][j]);
            System.out.println();
        }
    }
    
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        System.out.println("Enter matrix size: ");
        int size = in.nextInt();
        
        int[][] matrix = new int[size][];
        for (int i = 0; i < size; i++)
            matrix[i] = new int[size];
    
        fillDiagonally(matrix);
        
        System.out.println("Matrix is:");
        showMatrix(matrix);
    }
}