допоможіть переробити код з урахуванням класу або дайте хоть приклад на с++

#include <iostream>
#include <ctime>
using namespace std;

int main()
{
    setlocale(LC_ALL, "ukraine");
    cout << "\t\tСлід матриці\n";

    srand(time(NULL));
    int X = 0;
    int n = 0;
    cout << "введіть кількість стовпців і рядків: ";
    cin >> n; 
    int** a = new int* [n];
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            a[i] = new int[n];
        }
    }
    
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            a[i][j] = rand() % 100;
            cout << a[i][j] << " ";
            if (a[i] == a[j])
            {
                X += a[i][j];
            }
        }
        cout << endl;
    }
    cout << "слід матриці: " << X << endl;
    // Удаление массива
    for (int i = 0; i < n; i++)
    {
        delete[]a[i];
    }
    delete[] a;
    return 0;
}

дякую

Так

Тільки почали вчити ООП і класи.
(і бац таке завдання)
і до кінця ще не розумію, як з ним взаємодіяти

MMORPG, ось готовий код з використанням шаблону (template) :

// Matrix.h

#ifndef MATRIX_H
#define MATRIX_H

#include <exception>
#include <iostream>

template<typename T>
class Matrix {
    private:
        int n, m;
        T **a;
    public:
        Matrix(int n, int m) : n(n), m(m), a(nullptr) {
            a = new T*[n];
            if(!a) throw std::bad_alloc();
            for(int i(0); i < n; ++i) {
                T *ai(a[i] = new T[m]);
                if(!ai) {
                    for(--i; i >= 0; --i)
                        delete[] a[i];
                    delete[] a;
                    a = nullptr;
                    throw std::bad_alloc();
                }
                for(int j(0); j < m; j++)
                    ai[j] = 0;
            }
        }
        ~Matrix() {
            if(!a) return;
            for(int i(0); i < n; ++i)
                delete[] a[i];
            delete[] a;
            a = nullptr;
        }
        T *operator [](int i) {
            return (i >= 0 && i < n && a) ? a[i] : nullptr;
        }
        T const *operator [](int i) const {
            return (i >= 0 && i < n && a) ? a[i] : nullptr;
        }
        T get_trace() const {
            T trace(0);
            if(!a) return trace;
            int c = n < m ? n : m;
            for(int i(0); i < c; ++i)
                trace += a[i][i];
            return trace;
        }
        int get_n() const { return n; }
        int get_m() const { return m; }
        void init(T (*generator)()) {
            for(int i(0); i < n; ++i)
                for(int j (0); j < m; ++j)
                    a[i][j] = generator();
        }
};

using namespace std;

template<typename T>
ostream &operator <<(ostream &out, Matrix<T> const &matrix) {
    int n(matrix.get_n()), m(matrix.get_m()), i(0), j(0);
    T const *row(matrix[i]);
    out << "[\r\n  [ " << row[j];
    for(++j; j < m; ++j)
        out << ", " << row[j];
    out << " ]";
    for(++i; i < n; ++i) {
        row = matrix[i];
        out << ",\r\n  [ " << row[j = 0];
        for(++j; j < m; ++j)
            out << ", " << row[j];
        out << " ]";
    }
    out << "\r\n]";
    return out;
}

#endif

// Main.cpp

#include <ctime>
#include <iostream>
#include "Matrix.h"

int main(int argc, char const *argv[]) {
    Matrix<int> matrix(3, 5);
    srand(time(nullptr));
    matrix.init([]() -> int { return rand() % 10; });
    /*//
    matrix[0][0] = 2;
    matrix[1][1] = 3;
    matrix[2][2] = 4;
    //*/
    std::cout << "Matrix : " << matrix << "\r\n";
    std::cout << "Trace : " << matrix.get_trace() << endl;
    return 0;
}

А Matrix.cpp можна видалити.