Ex. 4.2 (2)

// ЗАДАНИЕ 4.2 (2)
// Дан двумерный числовой массив.
// Значения элементов главной диагонали возвести в квадрат.

#include <iostream>     // Для ВВОДА и ВЫВОДА

using namespace std;    // ПРОСТРАНСТВО ИМЁН

int main() {
    // Ввод размерма массива
    int size;
    cout << "Enter size matrix: ";
    cin >> size;

    int ** matrix = new int * [size];   // Объявление массива

    // Заполнение матрицы
    for (int i = 0; size > i; i++) {
        matrix[i] = new int [size];    // Объявление новой строки

        // Заполнение строки
        for (int j = 0; size > j; j++) {
            // Ввод значения элемента
            cout << "Enter matrix[" << i << "][" << j << "]: ";
            cin >> matrix[i][j];

            if (i == j) {
                // Если это элемент главной диагонали
                // то возводим в квадрат
                matrix[i][j] *= matrix[i][j];
            }
        }
    }

    cout << endl;

    // Вывод НОВОЙ МАТРИЦЫ
    cout << "New Matrix:" << endl;
    for (int i = 0; size > i; i++) {
        for (int j = 0; size > j; j++) {
            cout << matrix[i][j] << " ";
        }

        cout << endl;
    }

    return 0;
}