Найти в каждой строке наибольший элемент и поменять его местами с элементом главной диагонали

#include <iostream>
#include <cstdlib>
#include <ctime>
#include <iomanip>
using namespace std;
int main(void)
{
    srand(static_cast<int>(time(NULL)));
    const int size=5;
    int mat[size][size];
    int max_str=0,num_max_str=0,tmp;
    cout<<"Matrix:"<<endl;
    for(int i=0; i<size; i++)
    {
        for(int j=0; j<size; j++)
        {
            mat[i][j]=rand()%99;
            cout<<setw(4)<<mat[i][j]<<setw(4);
        }
       cout<<endl;
    }
    for(int i=0; i<size; i++)
    {
        max_str=mat[i][0];
        for(int j=0; j<size; j++)
        {
            if(max_str<mat[i][j])
            {
                max_str=mat[i][j];
                num_max_str=j;
            }
        }
        tmp=mat[i][num_max_str];
        mat[i][num_max_str]=mat[i][i];
        mat[i][i]=tmp;
    }
   cout<<endl<<"New matrix"<<endl;
   for(int i=0; i<size; i++)
   {
       for(int j=0; j<size; j++)
       {
           cout<<setw(4)<<mat[i][j]<<setw(4);
       }
      cout<<endl;
   }
  system("pause");
}