диагональный обход матрицы

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

void matrix( int n)
{
  int s=0, res=0;
  int arr[n][n];
  arr[0][0]=1;

   for (int j=1; j<n; j++)
      arr[0][j]=j+1+arr[0][j-1];

   for (int i=1; i<n; i++)
    for (int j=0; j<n; j++)
     {
       if (j!=n-1)
     arr[i][j]=arr[i-1][j+1]-1;
     else arr[i][j]=arr[i-1][j]+n-i;
     }

   for (int i=0; i<n; i++)
   {
     for (int j=0; j<n; j++)
      cout << arr[i][j] << " ";
      cout << endl;
   }

}
int main()
{
  int a=0;
  cin >> a;
  matrix (a);
}