Заполнение главной диагонали матрицы максимальными элементами

template <class TYPE> void Func(TYPE x[][N], int M)
{
    int max, imax, jmax;
 
    for(int k=0; k<M; k++)
    {
        imax = 0; jmax = 0;
 
        for(int i=0; i<M; i++)
        {
            for(int j=1; j<N; j++)
            {
                if(x[i][j] > x[imax][jmax] && (i != k && j != k))
                    x[imax][jmax] = x[i][j];
            }
        }
        swap(x[k][k], x[imax][jmax]);
    }
}

{
    int max, imax, jmax;
 
    for(int k=0; k<N; k++)
    {
        imax = 0; jmax = 0;
 
        for(int i=0; i<N; i++)
        {
            for(int j=1; j<N; j++)
            {
                if(x[i][j] > x[imax][jmax] && (i != k && j != k))
                {
                    imax = i;
                    jmax = j;
                    max = x[i][j];
                }
            }
        }
        swap(x[k][k], x[imax][jmax]);
    }
}

#include <ctime>
#include <cstdlib>
#include <iomanip>
#include <iostream>
using namespace std;
 
template <class T>
void Func(T ** arr, int size);
 
int main()
{
    int i, j;//Счётчики
    int m      = 0;//Будет содержать размерность матрицы
    int ** arr = 0;//Указатель на матрицу
    cout<<"Enter size of matrix : ";cin>>m;
    srand(time(0));//Устанавливаем нач значение в генераторе случ чисел
    arr = new int *[m];//Выделили память под указатели на строки матрицы
    cout<<"\tInput array\n";
    for(i = 0; i < m; i++)
    {
        arr[i] = new int[m];////Выделили память под элементы i-й строки матрицы
        for(j = 0; j < m; j++)
            cout<<setw(3)<<(arr[i][j] = rand()%(m*m))<<" ";
        cout<<endl;
    }
    Func(arr, m);//Преобразуем массив
    cout<<"\tTransformed array\n";
    for(i = 0; i < m; i++)
    {
        for(j = 0; j < m; j++)
            cout<<setw(3)<<arr[i][j]<<" ";
        cout<<endl;
    }
    return 0;
}
 
template <class T>
void Func(T ** arr, int size)
{
    int i, j, k;
    int imax = 0;
    int jmax = 0;
    for(k = 0; k < size; k++)
    {
        imax = (jmax = k);
        for(i = 0; i < size; i++)
        for(j = 0; j < size; j++)
        {
            
            if(i == j && i <= k) 
                continue;//Не рассматриваем предыдущие элементы на главной диагонали
            else
            {
                if(arr[imax][jmax] < arr[i][j])
                {
                    imax = i;
                    jmax = j;
                }
            }
        }
        swap(arr[k][k], arr[imax][jmax]);
    }
}