C++, След матрицы

Задача:

След матрицы – это сумма диагональных элементов этой матрицы. Напишите программу, которая принимает с клавиатуры число N, создает матрицу размера N на N, заполненную случайными числами, и выводит ее на экран. Затем вычислите след этой матрицы.

Для решения этой задачи пригодятся формулы диагоналей из задачи «Единичная матрица»

Я написал 2 варианта решения. Первый — в несколько циклов. В первом заполняем, во втором считаем. Во втором случае — один цикл, сразу заполняем и считаем.

Вариант 1:

#include <iostream>
#include <ctime>

using namespace std;

int main()
{
	srand(time(NULL));
	int n = 0; 
	int sum = 0;
	cin >> n; // Считываем с клавиатуры n
	int **a = new int* [n]; // Создаем массив указателей
	for (int i = 0; i < n; i++)
	{
		a[i] = new int [n]; // Создаем элементы
	}
	// А дальше работа как с обычным массивом. 
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			a[i][j] = rand() % 10;
			cout << a[i][j] << " "; // Вывести элементы на консольку
		}
		cout << endl; // Двумерный массив. Строка кончилась, переводим строку и на консоли
	}

	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			if (i == j || j==n-i-1)
				sum += a[i][j];
		}
	}

	cout << "sum = " << sum << endl;

	// Удаление массива
	for (int i = 0; i < n; i++)
	{
		delete[]a[i]; // Удаляем каждый элемент
	}
	delete [] a; // А потом массив
	return 0;
}

Вариант 2:

#include <iostream>
#include <ctime>

using namespace std;

int main()
{
	srand(time(NULL));
	int n = 0; 
	int sum = 0;
	cin >> n; // Считываем с клавиатуры n
	int **a = new int* [n]; // Создаем массив указателей
	for (int i = 0; i < n; i++)
	{
		a[i] = new int [n]; // Создаем элементы
	}
	// А дальше работа как с обычным массивом. 
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			a[i][j] = rand() % 10;
			if (i == j || j==n-i-1)
				sum += a[i][j];
			cout << a[i][j] << " "; // Вывести элементы на консольку
		}
		cout << endl; // Двумерный массив. Строка кончилась, переводим строку и на консоли
	}

	cout << "sum = " << sum << endl;

	// Удаление массива
	for (int i = 0; i < n; i++)
	{
		delete[]a[i]; // Удаляем каждый элемент
	}
	delete [] a; // А потом массив
	return 0;
}

Смысл в любом из случаев в том, что в самом начале объявляется переменная sum, а дальше к ней прибавляется значение элемента, если он на диагонали.