Напечатать квадрат матрицы с наименьшей суммой диагональных элементов

#include "stdafx.h"
#include <iostream>
#include <random>
#include <iomanip>
 
const int n = 6;
 
void PrintMatrix(double a[][n], const char* s) {
    std::cout << s << "\n";
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++)
            std::cout << std::setprecision(3) << a[i][j] << "\t";
        std::cout << "\n";
    }
    std::cout << "\n";
}
 
void MatrixSqr(double (&a)[n][n]) {
    double c[n][n];
    for (int i = 0; i < n; i++) {
        for (int l = 0; l < n; l++) {
            double s = 0;
            for (int j = 0; j < n; j++) {
                s += a[i][j] * a[j][l];
            }
            c[i][l] = s;
        }
    }
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            a[i][j] = c[i][j];
}
 
int main()
{
    setlocale(LC_ALL, "rus");
    std::uniform_real_distribution<double> dist(-20, 20);
    std::default_random_engine gen;
    double a[n][n], b[n][n], sum_a = 0.0, sum_b = 0.0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            a[i][j] = dist(gen);
            b[i][j] = dist(gen);
        }
        sum_a += a[i][i];
        sum_b += b[i][i];
    }
    PrintMatrix(a, "Матрица A:");
    PrintMatrix(b, "Матрица B:");
    std::printf("Сумма диагонали А: %f\nСумма диагонали B: %f\n\n", sum_a, sum_b);
    if (sum_a < sum_b) {
        MatrixSqr(a);
        PrintMatrix(a, "Квадрат матрицы A:");
    }
    else {
        MatrixSqr(b);
        PrintMatrix(b, "Квадрат матрицы B:");
    }
    system("pause");
}