Сортировка диагоналей матрицы методом вставки - C++ - Ответ 9033961

#include <iostream>
#include <fstream>
#include <iomanip>
using namespace std;
 
ifstream in("input.txt");
ofstream out("out.txt");
 
void sort(int *a, int n)
{
    int temp;
    int i, j;
    for (i = 1; i<n; i++)
    {
        j = i;
        while (a[j]>a[j - 1] && j >= 1)
        {
            temp = a[j];
            a[j] = a[j - 1];
            a[j - 1] = temp;
            j--;
        }
    }
}
int main()
{
    int n, m, i, j;
    int a[4][4];
    in >> n >> m;
    for (i = 0; i<n; i++)
        for (j = 0; j<m; j++) in >> a[i][j];
 
    int *diag1;
    diag1 = new int[n + 1];
    for (i = n - 2; i>0; i--)
    {
        for (j = 0; j<n - i; j++) diag1[j] = a[j][i + j];
        sort(diag1, n - i);
        for (j = 0; j<n - i; j++) a[j][i + j] = diag1[j];
    }
    int *diag2;
    diag2 = new int[n + 1];
    for (i = 1; i<n - 1; i++)
    {
        for (j = i; j <= n - 1; j++) diag2[j - 1] = a[j][j - i];
 
        for (int k = 0; k<n - i; k++)
        {
            cout << diag2[k] << "\t";
        } cout << endl;
        sort(diag2, n - i);
        for (j = i; j <= n - 1; j++) a[j][j - i] = diag2[j];
    }
    delete diag1;
    delete diag2;
    out << n << '\t' << m << '\n';
    for (i = 0; i<n; i++)
    {
        for (j = 0; j<n; j++)
            out << setw(5) << a[i][j];
        out << '\n';
    }
    in.close(); out.close();
    system("PAUSE");
    return 0;
}