pokazivaci 24

/*Napišite program za unos i ispis elemenata cjelobrojne kvadratne matrice.
Dinamički alocirati memoriju za kvadratnu matricu. Funkcija treba vratiti
pokazivač na niz elemenata koji predstavlja niz elemenata sa glavne
dijagonale matrice. Elementima matrice u funkciji pristupati preko pokazivača.
Izvršite dealokaciju memorije.*/


#include <iostream>

using namespace std;

int *d(int **mat, int n);


int main()
{
    int n;
    cout<<"Unesi broj redaka i stupaca."<<endl;
    cin>>n;

    int **mat=new int*[n];
    for(int i=0; i<n; i++)
        mat[i]=new int[n];

    cout<<"Unos elemenata matrice:"<<endl;

    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            cin>>*(*(mat+i)+j);

    cout<<"Ispis:"<<endl;

    for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                cout<<*(*(mat+i)+j);
            }
            cout<<endl;
        }

    cout<<"Dijagonala: "<<endl;
    int *pok=d(mat, n);

    for(int i=0; i<n; i++)
        cout<<*(pok+i)<<" ";

    for(int i=0; i<n; i++)
        delete []mat[i];
    delete []mat;
    mat=0;

    delete []pok;
    pok=0;

    return 0;

}

int *d(int **mat, int n)
{
    int *p = new int[n];

    for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
               if(i==j)
                    *(p+i) = *(*(mat+i)+j);
            }
        }

    return p;


}