Квадратная матрица в виде одномерного массива

#include <iostream>
#include <cstdlib>
 
using namespace std;
 
int main()
{
    double *vector = NULL;
    double **matrix = NULL;
    int n;
    int i, j, k, t;
 
    cout << "Input n: ";
    cin >> n;
 
    vector = new double [n * n];
 
    matrix = new double * [n];
 
    if (vector == NULL || matrix == NULL)
    {
        cout << "Error!";
        return -1;
    }
 
    for (i = 0; i < n; i++)
    {
        matrix[i] = new double [n];
 
        if (matrix[i] == NULL)
        {
            cout << "Error!";
            return -1;
        }
    }
 
    for (i = 0; i < n * n; i++)
    {
        cout << "Input vector[" << i << "]: ";
        cin >> vector[i];
    }
 
    for (i = 0; i < n; i++)
        for (j = 0; j < n; j++)
            matrix[j][i] = vector[i * n + j];
 
    for (i = 0; i < n; i++)
    {
        for (j = 0; j < n; j++)
            cout << matrix[i][j] << "  ";
 
        cout << endl;
    }
 
    cout << endl << endl;
 
    for (i = 0, j = 0, k = 0; i < n; i++, k++)
    {
        j = 0;
        t = k;
 
        while (t-- > 0)
        {
            cout << "   ";
            j++;
        }
 
        for ( ; j < n; j++)
        {
            cout << matrix[i][j] << "  ";
        }
 
        cout << endl;
    }
 
    for (i = 0; i < n; i++)
        delete [] matrix[i];
 
    delete [] matrix;
 
    delete [] vector;
 
    cout << endl;
 
    system("pause");
    return 0;
}