Напечатать квадрат той матрицы, в которой наименьший след - C++ - Ответ 1321005

#include <iostream>
 
int sled (int ** a, int n)
{
    int sum = 0;
    for (int i = 0; i < n; i++)
        sum += a [i][i];
    return sum;
}
 
int ** mult (int ** a1, int ** a2, int n)
{
    int s = 0;
    int ** res = new int * [n];
    for (int i = 0; i < n; i++)
        res [i] = new int [n];
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            s = 0;
            for (int k = 0; k < n; k++)
                s += a1 [i][k] * a2 [k][j];
            res [i][j] = s;
        }
    }
 
    return res;
}
 
void printMatrix (int ** a, int n, std::ostream & out)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            cout << a [i][j] << " ";
        }
        cout << endl;
    }
}
 
int main()
{
    int N = 3;
    int ** a1 = new int * [N];
    int ** a2 = new int * [N];
 
    for (int i = 0; i < N; i++)
    {
        a1 [i] = new int [N];
        a2 [i] = new int [N];
    }
 
    std::cout << "Matrix 1\n";
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            cin >> a1 [i][j];
 
    std::cout << "\nMatrix 2\n";
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            cin >> a2 [i][j];
 
    if (sled (a1, N) > sled (a2, N))
        a1 = mult (a1, a1, N);
    else
        a1 = mult (a2, a2, N);
 
    std::cout << std::endl;
    printMatrix (a1, N, std::cout);
    return 0;
}