popravni-sept-2014-zad8-Alokacija

#include <iostream>
#include <stdexcept>
#include <algorithm>


using namespace std;

template<typename TipPok, typename TipEl>
void Alociraj (TipPok &a, int m, int n, TipEl v, bool mod ) {

    if (!mod) {
try {
    TipPok mat=nullptr;
    mat=new TipEl *[m];
    for (int i=0; i<m; i++)
        mat[i]=nullptr;

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

    for (int i=0; i<m; i++) {
        for (int j=0; j<n; j++)
            mat[i][j]=v;
    }
    a=mat;

}
 catch(bad_alloc) {
throw "Greska prilikom alokacije.";
}}
else {
    try { TipPok mat=nullptr;
        mat=new TipEl  *[m];
        mat[0]=new TipEl [m*n];
        for (int i=1; i<=m; i++) mat[i]=mat[i-1]+n;

    for (int i=0; i<m; i++) {
        for (int j=0; j<n; j++)
        mat[i][j]=v;
    }
    a= mat;
    }
    catch(bad_alloc) {
throw "Greska prilikom alokacije.";
}}
}

int main () {
    try {
        int **pok;
        Alociraj(pok,3,3,0, true);

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

    delete [] pok[0];
    delete[] pok;
    }
    catch (const char poruka[]) {
    cout<<poruka;
    }
    return 0;
}