Упорядочить элементы матрицы по убыванию элементов (направление сверху-вниз, слева - направо)

import java.util.Arrays;
import java.util.Collections;
import java.util.Scanner;
 
public class Main {
 
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        System.out.println("Enter a matrix size:");
        System.out.print("n = ");
        int n = in.nextInt();
        System.out.print("m = ");
        int m = in.nextInt();
        int[][] a = new int[n][m];
        Integer[] tmp = new Integer[n*m];
        int k = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                a[i][j] = in.nextInt();
                tmp[k++] = a[i][j];
            }
        }
        Arrays.sort(tmp, Collections.reverseOrder());
        k = 0;
        for (int j = 0; j < m; j++) {
            for (int i = 0; i < n; i++) {
                a[i][j] = tmp[k++];
            }
        }
        System.out.println("Output of the program:");
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                System.out.printf("%d ", a[i][j]);
            }
            System.out.println();
        }
    }
}