Найти максимальный элемент каждой строки двумерного массива и соответственно заменить их элементами в главной

#include <algorithm>
#include <array>
#include <iostream>
 
int main()
{
    constexpr size_t N = 4;
    std::array<std::array<int, N>, N> arr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6};
 
    for (const auto &l : arr)
    {
        for (const auto &x : l)
            std::cout << x << " ";
        std::cout << std::endl;
    }
    std::cout << std::endl;
 
    for (size_t i = 0; i < N; i++)
    {
        std::swap(arr[i][i], *std::max_element(arr[i].begin(), arr[i].end()));
        for (const auto &x : arr[i])
            std::cout << x << " ";
        std::cout << std::endl;
    }
}

#include "stdafx.h"
#include <iostream>
using namespace std;
 
int n;
 
void Output(int** a) {
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++)
            cout << a[i][j] << "\t";
        cout << endl;
    }
}
 
int main()
{
    cout << "n = "; cin >> n;
    int** a = new int*[n];
    for (int i = 0; i < n; i++)
        a[i] = new int[n];
 
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            a[i][j] = rand() % 101 - 50;
    cout << "Matrix:\n"; Output(a);
    for (int i = 0; i < n; i++) {
        int max = a[i][0];
        for (int j = 1; j < n; j++)
            if (a[i][j] > max)
                max = a[i][j];
        a[i][i] = max;
    }
    cout << "Result:\n"; Output(a);
 
    for (int i = 0; i < n; i++)
        delete[] a[i];
    delete a;
    system("pause");
    return 0;
}

std::swap(arr[i][i], *std::max_element(arr[i].begin(), arr[i].end()));

arr[i][i] = *std::max_element(arr[i].begin(), arr[i].end());