Заполнение матрицы по правилу

#include <iostream>
 
    using namespace std;
 
int main() {
    int n, i, j;
    cout << "Enter a matrix size:\n";
    cout << "n = ";
    cin >> n;
    int ** a = new int * [n];
    for (int i = 0; i < n; i++) {
        a[i] = new int[n];
    }
    for (int p = n - 1; p >= 1 - n; p--) {
        if (p > 0) {
            i = n - p;
            for (j = 0; j < n - p; j++) {
                --i;
                a[i][j] = 0;
            }
 
        } else if (p < 0) {
            i = n;
            for (j = abs(p); j < n; j++) {
                --i;
                a[i][j] = 2;
            }
        } else {
            for (int i = 0; i < n; i++) {
                a[n-i-1][i] = 1;
            }
        }
    }
    cout << "Output of the program:\n";
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cout << a[i][j] << " ";
        }
        cout << "\n";
    }
    for (int i = 0; i < n; i++) {
        delete [] a[i];
    }
    delete [] a;
    system("pause");
    return 0;
}

#include <iostream>
#include <ctime>
using namespace std;
 
int main()
{    
    int n;
    cout <<"n="; cin >>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++)
    {
      if (i==n-1-j) a[i][j]=1;
      else if (i>n-1-j) a[i][j]=2;
          else a[i][j]=0;
      cout <<a[i][j]<<" ";
    }
    cout <<endl;
    }
 
    for (int i = 0; i < n; i++)
    delete [] a[i];
    delete [] a;
 
system("pause");
return 0;
}