Матрицу в строку по вторичным диагоналям

import java.util.PrimitiveIterator.OfInt;
import java.util.stream.IntStream;
 
public class Test {
    public static void main(String[] args) {
        int[][] matrix = new int[5][5];
        
        fillMatrix(matrix);
        printMatrix(matrix);
        
        System.out.println("\n" + doTheThing(matrix));
    }
    
    private static String doTheThing(int[][] matrix) {
        StringBuilder sb = new StringBuilder();
        
        int i, j, k;
        for (i = 0; i < matrix.length; i++) {
            for (j = i, k = 0; k <= i; j--, k++) {
                sb.append(matrix[j][k]).append(' ');
            }
        }
        
        for (i = 1; i < matrix.length; i++) {
            for (j = matrix.length - 1, k = i; k < matrix[0].length; j--, k++) {
                sb.append(matrix[j][k]).append(' ');
            }
        }
        
        return sb.toString();
    }
    
    private static void fillMatrix(int[][] matrix) {
        OfInt iter = IntStream.iterate(1, i -> i + 1).iterator();
        
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                matrix[i][j] = iter.nextInt();
            }
        }
    }
    
    private static void printMatrix(int[][] matrix) {
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                System.out.printf("%3d", matrix[i][j]);
            }
            System.out.println();
        }
    }
}

import java.util.Scanner;
 
public class Main  {
 
    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        System.out.println("Введите размер матрицы: ");
        int n = in.nextInt();
 
        int[][] matrix = new int[n][n];
        matrix(n, matrix);
        outRandomMatrix(matrix);
 
        int[] newArray = new int [n*n];
        newArray(n, matrix, newArray);
        outArray(newArray);
    }
 
    private static int[][] matrix(int n, int[][] matrix) {                     
    for (int i = 0; i < matrix.length; i++) {
    for (int j = 0; j < matrix[i].length; j++) {
    matrix[i][j] = (int) (Math.random() * 100);
      }
      }
    return (matrix);
    }
 
    private static void outRandomMatrix(int[][] matrix){
    System.out.println("Матрица рандомных чисел:");
    for (int i = 0; i < matrix.length; i++){
    for (int j = 0; j < matrix[i].length; j++){
    System.out.print(matrix[i][j] + " ");
      }
        System.out.println();
      }
    }
 
    private static int[] newArray(int n, int[][]matrix, int[] newArray) {
 
    for (int i = 0; i < matrix.length; i++) {
    for (int j = 0; j < matrix[i].length; j++) {
       newArray[matrix.length * i + j] = matrix[i][j];
     }
     }
     return (newArray);
    }
 
     private static void outArray(int[] newArray){
        System.out.println("Матрица преобразована в масив:");
        for (int i = 0; i < newArray.length; i++){
            System.out.print(newArray[i] + " ");
        }
        }
 
 
    private static void InputDialog(String s) {
    }
}