p78e17(2)

#include <iostream>
#include <cmath>
#include <fstream>
#include <vector>

using namespace std;

template < typename create, typename create1 >
create* CreateFun (int &n, ifstream &inp)
{
        inp >> n;
        create *mas = new create [n];
        for (int i = 0; i < n; i++)
        {
                mas[i] = new create1 [n];
                for ( int j = 0; j < n; j++)
                        inp >> mas[i][j];
        }
        return mas;
}

template < typename print >
void PrintFun (print *mas, int &n)
{
	for (int i = 0; i < n; i++)
     {
        for ( int j = 0; j < n; j++)
			cout << mas[i][j] << " ";
        cout << endl;
     }
        cout << endl;
}


void sort_ab (vector<int*> &a)
{
	int idx;
	int key;
	for(int i = 0; i < a.size(); i++)
	{
		idx = i;
		key = *a[i];
		for(int j = i + 1; j < a.size(); j++)
			if ( *a[j] < key)
			{
				idx = j;
				key = *a[j];
			}
		*a[idx] = *a[i];
		*a[i] = key;
	}
}

void sort_un (vector<int*> &a)
{
	int idx;
	int key;
	for(int i = 0; i < a.size(); i++)
	{
		idx = i;
		key = *a[i];
		for(int j = i + 1; j < a.size(); j++)
			if ( *a[j] > key)
			{
				idx = j;
				key = *a[j];
			}
		*a[idx] = *a[i];
		*a[i] = key;
	}
}

int main(void)
{
	ifstream inp ("input.txt");
	int n;
	int **mas = CreateFun <int*,int> (n,inp);
	PrintFun <int*> (mas,n);


    for (int k = 0; k <= n; k++)
	{
        vector<int*> aboveDiagonal;
        vector<int*> underDiagonal;

        for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
			{
                if (i==j-k) aboveDiagonal.push_back(&mas[i][j]);
                if (i==j+k) underDiagonal.push_back(&mas[i][j]);
			}
        if (! (aboveDiagonal.size() == n) && ! (underDiagonal.size() == n))
		{
           sort_ab(aboveDiagonal);
           sort_un(underDiagonal);
        }
    }
    PrintFun <int*> (mas,n);
}