Отсортировать элементы двухмерного массива находящиеся на одной диагонали

[S.Petrov]

Есть двухмерный массив n x m строк вида:

0 | 5 | 10 | 15
1 | 6 | 11 | 16
2 | 7 | 12 | 17
3 | 8 | 13 | 18
4 | 9 | 14 | 19

Нужно преобразовать в массив содержащий список элементов, которые находятся на одной диагонали, например: [0], [1, 5], [2, 6, 10], [3, 7, 11, 15], [4, 8, 12, 16], [9, 13, 17], [14, 18], [9].

Подскажите, пожалуйста, как такое можно сделать или где можно посмотреть, решение.