Найти максимальный элемент в главной диагонали матрицы и поменять местами соответствующие ему строку и столбец

#include <iostream>
#include <iomanip>
#include <time.h>
 
int main()
{
    srand(time(NULL));//рандом по вермени
    const int n=5+1;//размер массива nxn
    int massiv[n][n]={0};//Все элементы равны 0
    int max=0, I=0, J=0;
    //заполнение массива произвольными числами(рандом) и вывод на экран
    for(int i=1;i<n;i++)
    {
        for(int j=1;j<n;j++)
        {
            massiv[i][j]=1+rand()%9;
            std::cout << massiv[i][j] << " ";
        }
        std::cout << std::endl;
    }
    //поиск
    for(int i=1, j=1;((i<n)&&(j<n));i++,j++)
    {
        if(massiv[i][j]>max)
        {
            max=massiv[i][j];
            I=i;J=j;
        }
    }
    std::cout << "Max element= " << max << " s indexami " << I << " "<< J << std::endl;
    //замена строки столбцом и стройкой столбец
    for(int i=1;i<n;i++)
    {
        for(int j=1;j<n;j++)
        {
            if((j==J)||(i==I))
                std::cout << massiv[j][i] << " ";
            else
                std::cout << massiv[i][j] << " ";
        }
        std::cout << std::endl;
    }
 
    return 0;
}

#include <iostream>
#include <string.h>
#include <time.h>
 
 
using namespace std;
int main (void) {
    srand((unsigned)time(NULL));
    int n=3;
    int tmpij = 0,
        tmpmat;
 
    int mat[n][n];
 
 
    for (int i = 0; i < n;  i++){
        for (int j = 0; j < n; j++) {
            mat[i][j] = rand()%10;
        }
    }
 
 
    for (int i = 0; i < n;  i++){
         cout <<  endl;
        for (int j = 0; j < n; j++) {
            cout << mat[i][j] << " ";
        }
    }
 
    int max = mat[0][0];
    cout << "\n================\n";
    for (int i = 0; i < n;  i++){
            {
                if (mat[i][i] >= max)
                {
                     max = mat[i][i];
                     tmpij = i;
                }
            }
    }
    cout << "\nmax element = " << max << "\non position : [" << tmpij+1 << "][" << tmpij+1 << "]" << endl;
 
    for (int i = 0; i < n; i++) {
 
        tmpmat = mat[i][tmpij];
        mat[i][tmpij] = mat[tmpij][i];
        mat[tmpij][i] = tmpmat;
    }
 
     for (int i = 0; i < n;  i++){
         cout <<  endl;
        for (int j = 0; j < n; j++) {
            cout << mat[i][j] << " ";
        }
    }
 
    cin.get();
    return 0;
}